small tips

发表于 更新于 分类于 阅读次数： Valine：
本文字数： 283 阅读时长 ≈ 1 分钟
  1. 在pytorch中，如果用交叉熵损失，则模型的最后一层不能是nn.Softmax()，因为在交叉熵损失中已经内置了nn.Softmax()，用两次会造成损失收敛缓慢甚至不收敛；
  2. 给文件起名字的时候不要与安装的包的名字相同，否则导包时可以会导本地同名的文件，调用包里的方法就会报，没有发现该模块；
  3. 安装完augly，发现导包失败，pip uninstall python-magic; pip install python-magic-bin==0.4.14；
  4. 训练vgg16时，学习率不能设置太大，比如设置为0.001时，就会造成损失函数不下降，模型不收敛，可以设置为0.0001；