Pytorch RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn

lxztju / pytorch_classification

利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码

MIT License

1.38k stars 339 forks source link

Open LeBron-Jian opened 3 years ago

LeBron-Jian commented 3 years ago

在训练train.py 时，需要加上 loss.requiresgrad(True) 不然会报错 RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn

请问作者做分类，有遇到这个问题吗？

AilvenLiu commented 2 years ago

兄弟，解决了吗？