在执行resnet50_trainer.py时,出现如下错误:
Traceback (most recent call last):
File "resnet50_trainer.py", line 593, in <module>
main()
File "resnet50_trainer.py", line 589, in main
Train(args)
File "resnet50_trainer.py", line 429, in Train
shared_model=args.use_cpu,
TypeError: Parallelize() got an unexpected keyword argument 'shared_model'
原因为data_parallel_model中的Parallelize接口更新,增加了shared_model参数,样例中与源码不对应,所以出现这个错误。
解决方法:去掉shared_model=args.use_cpu参数,建议直接使用源码中example的代码
同样类型错误还有如下:解决方案一样
File "resnet50_trainer.py", line 339, in create_resnet50_model_ops
fp16_data=True if args.dtype == 'float16' else False,
TypeError: create_resnet50() got an unexpected keyword argument 'fp16_data'