iammelon的博客

问题原因： 我的电脑上cuda版本是8.0，但是第一次安装的是cuda10.1版本的pytorch，版本不一致导致的，应该安装与本机cuda版本对应的pytorch 解决办法： 在pytorch官网有对应不同cuda版本的pytorch 安装命令，网址是：https://pytorch.org/get-started/previous-versions/ 我的是8.0版本的cuda, 所以命令行是...

for inputs in dataLoader，for (inputs,) in dataLoader得到的inputs是不一样的。 for inputs in dataLoader: inputs = [tensor[1,2]], outputs=model(inputs[0]) for (inputs, ) in dataLoader: inputs = tensor[1,2], outpu...

训练完train_datasets之后，model要来测试样本了。在model(test_datasets)之前，需要加上model.eval(). 否则的话，有输入数据，即使不训练，它也会改变权值。这是model中含有batch normalization层所带来的的性质。 在做one classification的时候，训练集和测试集的样本分布是不一样的，尤其需要注意这一点。 ...