pytorch_classification
pytorch_classification copied to clipboard
利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码
你好,想问一下,出现ModuleNotFoundError: No module named 'cfg'需要怎么解决呀?谢谢
是什么的特征值?怎么求到这个2816的?
我使用resnet101网络模型,train和predict都可以正常运行,但运行cnn_ml出错了,大佬可以解答下吗?
RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘
AttributeError: 'VGG' object has no attribute 'fc'
 这一块数据集导入,是不是有点问题呀,index是什么东西呀,label也读不出来。一直显示下面的错误 
当自己尝试使用resnet50或者moblienetv2时都会报如下错误: `Traceback (most recent call last): File "train.py", line 147, in images, labels = next(batch_iterator) File "/home/xuhaowang/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 363, in __next__ data = self._next_data() File "/home/xuhaowang/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 403, in _next_data...
请教大佬这个bug应该怎么解决呢  另外,在train_kd.py中,定义优化器与损失函数那里,model是标红的,直接运行,显示没有定义; 当用teacher_model,student_model代替model时,显示下面这个bug。 optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=cfg.LR) optimizer = optim.SGD(model.parameters(), lr=cfg.LR, momentum=cfg.MOMENTUM, weight_decay=cfg.WEIGHT_DECAY)  抱拳
cnn_svm中的 get_test_transform 并未定义,是不是写错了?