关键代码
model_dir=os.path.join(run_dir,'model')
if not os.path.exists(model_dir):
os.mkdir(model_dir)
saver=tf.train.Saver()
model_name='ckp-01000'#导入前缀是该字符的模型
model_path=os.path.join(model_dir,model_name)
关键代码
model_dir=os.path.join(run_dir,'model')
if not os.path.exists(model_dir):
os.mkdir(model_dir)
saver=tf.train.Saver()
model_name='ckp-01000'#导入前缀是该字符的模型
model_path=os.path.join(model_dir,model_name)