为了保证代码能够复现,需要固定所有可能的随机数
import torch
import numpy as np
import random
def seed_everywhere(seed):
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed) # if you are using multi-GPU.
np.random.seed(seed) # Numpy module.
random.seed(seed) # Python random module.
最后
以上就是激情纸鹤最近收集整理的关于python设置随机数种子(numpy,pytorch,random)的全部内容,更多相关python设置随机数种子(numpy内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复