分类：代码段

2020
11-19

Pytorch实验常用代码段汇总

1.大幅度提升Pytorch的训练速度device=torch.device("cuda"iftorch.cuda.is_available()else"cpu")torch.backends.cudnn.benchmark=True但加了这一行，似乎运行结果不一样了。2.把原有的记录文件加个后缀变为.bak文件，避免直接覆盖#fromco-teachingtraincodetxtfile=save_dir+"/"+model_str+"_%s.txt"%str(args.optimizer)##goodjob！nowTime=datetime.datetime.now().strftime('%Y-%m-%d-%H:%M:%S... 继续阅读 >