Skip to content
Snippets Groups Projects
Commit a37ecacd authored by Fabrice Daian's avatar Fabrice Daian
Browse files

update

parent 6bd1c523
No related branches found
No related tags found
No related merge requests found
...@@ -185,7 +185,7 @@ def train(d_model, g_model, gan_model, dataset, output_path, val_dataset = None, ...@@ -185,7 +185,7 @@ def train(d_model, g_model, gan_model, dataset, output_path, val_dataset = None,
history[2].append(ssim) history[2].append(ssim)
print("Epoch", current_epoch) print("Epoch", current_epoch)
# print('Val> mse[%.3e], ssim[%.3e]' % (mse, ssim)) # print('Val> mse[%.3e], ssim[%.3e]' % (mse, ssim))
print("Val> mse[",str(mse),"], ssim[",str(ssim),"]" % (mse, ssim)) print("Val> mse[",str(mse),"], ssim[",str(ssim),"]")
cur_patience-=1 cur_patience-=1
val_loss = mse/((ssim+1)/2) # scaling ssim on 0,1 instead of -1,1 : trouble when ssim is <0 val_loss = mse/((ssim+1)/2) # scaling ssim on 0,1 instead of -1,1 : trouble when ssim is <0
with open(output_path+"/log.txt", 'a') as file: with open(output_path+"/log.txt", 'a') as file:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment