Loading

摘要: torch.manual_seed(seed) # 为CPU设置随机种子 torch.cuda.manual_seed(seed) # 为当前GPU设置随机种子 torch.cuda.manual_seed_all(seed) # if you are using multi-GPU,为所有GPU设 阅读全文
posted @ 2021-11-19 00:28 Guang'Jun 阅读(316) 评论(0) 推荐(0)