问题:训练中断后无法从checkpoint处恢复训练。
解决:
这个out_dir会导致load failed,因为有时间参数,load checkpoint时是当前时间参数,不是上次时间参数,导致failed,删了就好。
加载预训练模型:
main.py第123行左右:
if FLAGS.train:
dcgan.train(FLAGS)
else:
#load_success, load_counter = dcgan.load(FLAGS.checkpoint_dir)
load_success, load_counter = dcgan.load(r'./out\1 - data - celebA - x108.z100.uniform_signed.y64.b64\checkpoint')
if not load_success:
raise Exception("Checkpoint not found in " + FLAGS.checkpoint_dir)
修改dcgan.load后的路径,改为之前保存的checkpoint路径
修改生成图片的数量:修改main.py中的 "generate_test_images"参数
utils.py第197行左右
elif option == 1:
values = np.arange(0, 1, 1./config.batch_size)
for idx in xrange(dcgan.z_dim):
print(" [*] %d" % idx)
第三行改为for idx in xrange(config.generate_test_images):