未验证 提交 a9f35981 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: GitHub

fix fp16 config for dyg (#1052)

上级 a99de6a7
......@@ -27,6 +27,7 @@ AMP:
Arch:
name: ResNet50
class_num: 1000
input_image_channel: *image_channel
# loss function config for traing/eval process
Loss:
......
......@@ -19,6 +19,7 @@ Global:
Arch:
name: SE_ResNeXt101_32x4d
class_num: 1000
input_image_channel: *image_channel
# loss function config for traing/eval process
Loss:
......
......@@ -264,11 +264,6 @@ def build(config,
# data_format should be assigned in arch-dict
input_image_channel = config["Global"]["image_shape"][
0] # default as [3, 224, 224]
if input_image_channel != 3:
logger.warning(
"Input image channel is changed to {}, maybe for better speed-up".
format(input_image_channel))
config["Arch"]["input_image_channel"] = input_image_channel
model = build_model(config["Arch"])
out = model(feeds["data"])
# end of build model
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册