about train

[benyufly]

大佬，我不知道是环境配置的问题还是什么，我训练自己的数据集出现下面的错误

[zhangbaijin]

不好意思这么晚回复，环境跟SCI的环境一样就可以，另外要配置SAM所需要的依赖包，训练的命令是python train.py --arch WithoutCalNet --batchsize 10

[Dreamer-LRZ]

Hello, when I run train.py also appeared the following error : AttributeError : ' NoneType ' object has no attribute ' data ', what is the cause of the problem？

[zhangbaijin]

这个问题似乎是在加载预训练模型的时候，反向传播grad为0，可以移步我的github，更新最新的train代码