深度学习框架的一切收获(PyTorch、Tensorflow、Gluon)
Pytorch篇:
随机种子设定
import torch
import numpy as np
import random
import torch.backends.cudnn as cudnn
seed = 20
print('random seed: {}'.format(seed))
def setup_seed(seed):
torch.manual_seed(seed)...
原创
2019-06-05 20:59:58 ·
763 阅读 ·
0 评论