实验记录
文章平均质量分 58
H u
平平无奇的大学生实验记录工具博
展开
-
pytorch加载模型继续训练时optimizer参数被加载到cpu
pytorch加载模型位置原创 2022-08-11 23:26:56 · 1253 阅读 · 2 评论 -
Tensoeflow版本升级为2.0后运行1.0代码
Tensoeflow版本升级为2.0后运行1.0代码最近在跑一个用tensorflow1.0版本写的GAN,但是学校GPU平台默认的环境是tensorflow-gpu 2.0,2.0版本和1.0版本差别还是很大的,记录一下为了跑通代码调环境的过程…在查阅海量!!!资料后发现要跑1.0版本代码主要可以用两种方式:添加语句这种方法很简单,只需要在原代码中将import tensorflow as tf语句改为import tensorflow.compat.v1 as tftf.disable_v2原创 2021-02-03 23:14:49 · 2310 阅读 · 5 评论