高大石头

文章
5
资源
0
加入时间
2年10月21天

pytorch load state dict_pytorch训练trick

pytorch随机种子pytorch随机种子是随机初始化的,如果想复现一个比较好的结果,可以设置固定随机种子。其中cudnn打开可以提高计算效率,但是会影响每次复现结果。另外如果图像预处理的时候用了随机切割,随机翻转需要对python的随机数生成器进行固定。#增加运行效率torch.backends.cudnn.benchmark = False # if benchmark=Tr...