search

lxztju / pytorch_classification

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

AttributeError: Can't pickle local object 'get_random_eraser.<locals>.eraser' #30

Open hs756149694 opened 3 years ago

hs756149694 commented 3 years ago

我运行train.py文件时，碰到这个错误，有人解决过吗?

lxztju commented 3 years ago

注释掉dataset.py的第45行