当前位置 主页 > 网站技术 > 代码类 > 最大化 缩小

    Pytorch在dataloader类中设置shuffle的随机数种子方式

    栏目:代码类 时间:2020-01-14 12:09

    如题:Pytorch在dataloader类中设置shuffle的随机数种子方式

    虽然实验结果差别不大,但是有时候也悬殊两个百分点

    想要复现实验结果

    发现用到随机数的地方就是dataloader类中封装的shuffle属性

    查了半天没有关于这个的设置,最后在设置随机数种子里面找到了答案

    以下方法即可:

    def setup_seed(seed):
       torch.manual_seed(seed)
       torch.cuda.manual_seed_all(seed)
       np.random.seed(seed)
       random.seed(seed)
       torch.backends.cudnn.deterministic = True
    # 设置随机数种子
    setup_seed(20)

    以上这篇Pytorch在dataloader类中设置shuffle的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持IIS7站长之家。

    下一篇:没有了