您的位置:首页 > Web前端

caffe2分布式样例resnet50_trainer.py运行时出现参数shared_model错误

2018-01-30 14:48 369 查看
在执行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'
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  caffe2 分布式
相关文章推荐