在使用PyTorch训练SSD模型时，遇到了几个常见问题：1) 在train.py中更新visdom plot时遇到AssertionError，通过调整代码结构解决了；2) numpy创建ndarray时遇到VisibleDeprecationWarning，建议降低numpy版本或指定dtype=object；3) 使用volatile时收到UserWarning，需用torch.no_grad()替代；4) size_average和reduce参数被弃用，推荐使用reduction='sum'代替。
1. ERROR : assert win is not None, 'Must define a window to update' AssertionError: Must define a wind（train.py中修改)
上图是修改完后的样子，将这段函数上面代码中的epoch+=1挪到 “if args.visdom:" 函数下，并把update_vis_plot括号中的内容替换成图片中的内容，问题解决，具体可参考https://github.com/amdegroot/ssd.pytorch/issues/234
2. VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.
/home/bb622/ssd.pytorch-master/utils/augmentations.py:239: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of 
最低0.47元/天 解锁文章
8649
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
