pytorch load_state_dict 参数却未载入模型

训练时使用该随机种子代码后，在预测代码部分也需要放上同样的随机种子（注：我的训练和预测代码是在两个不同的.py文件）

SEED=1
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed_all(SEED)

不在预测代码部分加上上面的代码，权重文件就会失效。