未验证 提交 1bf72647 编写于 作者: Z Zhen Wang 提交者: GitHub

fix core dumped during the test stage of ResNet50 AMP. (#4646)

上级 cc6d8757
...@@ -194,7 +194,7 @@ def build(settings, mode='train'): ...@@ -194,7 +194,7 @@ def build(settings, mode='train'):
pipe, ['feed_image', 'feed_label'], pipe, ['feed_image', 'feed_label'],
size=len(pipe), size=len(pipe),
dynamic_shape=True, dynamic_shape=True,
fill_last_batch=False, fill_last_batch=True,
last_batch_padded=True) last_batch_padded=True)
file_list = os.path.join(file_root, 'train_list.txt') file_list = os.path.join(file_root, 'train_list.txt')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册