From cde404662266a51d1740dcc5d4d02951cd8a9f33 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Fri, 9 Feb 2024 15:38:36 +0100
Subject: [PATCH] fix int
---
main.py | 22 +++++++++++-----------
1 file changed, 11 insertions(+), 11 deletions(-)
diff --git a/main.py b/main.py
index bff9cd6..bf1509e 100644
--- a/main.py
+++ b/main.py
@@ -243,17 +243,17 @@ def run_int(epochs, eval_inter, save_inter, model, data_train, data_val, optimiz
train_int(model, data_train, e, optimizer, criterion, metric, wandb=wandb)
if e % eval_inter == 0:
loss = eval_int(model, data_val, e, criterion, metric, wandb=wandb)
- # if loss < best_loss:
- # best_epoch = e
- # if wandb is not None:
- # save(model, optimizer, epochs, 'model_int' + wandb + '.pt')
- # else:
- # save(model, optimizer, epochs, 'model_int.pt')
- # if wandb is not None:
- # model_final = load('model_int' + wandb + '.pt')
- # else:
- # model_final = load('model_int.pt')
- # print('Best epoch : ',e)
+ if loss < best_loss:
+ best_epoch = e
+ if wandb is not None:
+ save(model, optimizer, epochs, 'model_int' + wandb + '.pt')
+ else:
+ save(model, optimizer, epochs, 'model_int.pt')
+ if wandb is not None:
+ model_final = load('model_int' + wandb + '.pt')
+ else:
+ model_final = load('model_int.pt')
+ print('Best epoch : ',e)
def main_rt(args):
--
GitLab