Toybrick

config的参数

LSC

注册会员

积分
142
楼主
发表于 2020-1-7 14:20:54    查看: 7344|回复: 5 | [复制链接]    打印 | 只看该作者
  1. def config(self,
  2.                batch_size=100,
  3.                caffe_mean_file=None,
  4.                channel_mean_value=None,
  5.                dtype='float32',
  6.                epochs=-1,
  7.                force_gray=None,
  8.                input_fitting='scale',
  9.                input_normalization=None,
  10.                mean_file=None,
  11.                model_data_format='zone',
  12.                model_quantize=None,
  13.                optimize='Default',
  14.                quantized_dtype='asymmetric_quantized-u8',
  15.                quantized_moving_alpha=0.0,
  16.                quantized_algorithm='normal',
  17.                quantized_divergence_nbins=1024,
  18.                random_brightness=None,
  19.                random_contrast=None,
  20.                random_crop=None,
  21.                random_flip=None,
  22.                random_mirror=None,
  23.                reorder_channel=None,
  24.                restart=False,
  25.                samples=-1,
  26.                need_horizontal_merge=False,
  27.                deconv_merge=True,
  28.                quantized_hybrid=False,
  29.                output_optimize=0,
  30.                remove_tensorflow_output_permute=False):
  31.         """
  32.         Configs
  33.         :param batch_size: Batch size of input
  34.         :param caffe_mean_file: Caffe mean file
  35.         :param channel_mean_value: Channel mean value
  36.         :param dtype: Data type
  37.         :param epochs: Epochs
  38.         :param force_gray: Force image to gray
  39.         :param input_fitting:
  40.         :param input_normalization:
  41.         :param mean_file:
  42.         :param model_data_format:
  43.         :param model_quantize:
  44.         :param optimize:
  45.         :param quantized_dtype: quantize data type, currently support: asymmetric_quantized-u8, dynamic_fixed_point-8, dynamic_fixed_point-16
  46.         :param quantized_moving_alpha:
  47.         :param quantized_algorithm:
  48.         :param quantized_divergence_nbins:
  49.         :param random_brightness:
  50.         :param random_contrast:
  51.         :param random_crop:
  52.         :param random_flip:
  53.         :param random_mirror:
  54.         :param reorder_channel:
  55.         :param restart:
  56.         :param samples:
  57.         :param need_horizontal_merge: weather need merge horizontal layer, for inception serial models, it can optimize performance
  58.         :param deconv_merge: merge deconv with bn or bn+scale
  59.         :param quantized_hybrid:
  60.         :param output_optimize: optimize get output time
  61.         :param remove_tensorflow_output_permute: disable tensorflow model output permute, keep output format NCHW
  62.         :return: None
  63.         """
复制代码
请问,model_data_format、quantized_algorithm、optimize这三个参数是什么作用?除了默认值还有哪些值可选?


回复

使用道具 举报

LSC

注册会员

积分
142
沙发
 楼主| 发表于 2020-1-9 13:20:48 | 只看该作者
自己顶一下
回复

使用道具 举报

jefferyzhang

版主

积分
12925
板凳
发表于 2020-1-9 14:19:20 | 只看该作者
文档里有写
回复

使用道具 举报

zht

注册会员

积分
74
地板
发表于 2020-1-10 08:53:42 | 只看该作者
quantized_algorithm为:
kl_divergence
moving_average
normal (default)

其他有部分参数没有对外开放。
回复

使用道具 举报

LSC

注册会员

积分
142
5#
 楼主| 发表于 2020-1-10 15:36:29 | 只看该作者
zht 发表于 2020-1-10 08:53
quantized_algorithm为:
kl_divergence
moving_average

非常感谢,
回复

使用道具 举报

LSC

注册会员

积分
142
6#
 楼主| 发表于 2020-1-10 15:37:32 | 只看该作者

谢谢,不过文档有些不全,
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表