Skip to content

Commit 36f79f5

Browse files
authored
Update train.py
1 parent 9d02c84 commit 36f79f5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

train.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -334,8 +334,9 @@ def save_network(network, epoch_label):
334334
dir_name = os.path.join('./model',name)
335335
if not os.path.isdir(dir_name):
336336
os.mkdir(dir_name)
337-
copyfile('./train.py', dir_name+'/train.py')
338-
copyfile('./model.py', dir_name+'/model.py')
337+
#record every run
338+
copyfile('./train.py', dir_name+'/train.py')
339+
copyfile('./model.py', dir_name+'/model.py')
339340

340341
# save opts
341342
with open('%s/opts.json'%dir_name,'w') as fp:

0 commit comments

Comments
 (0)