From e0c511a5764d071c67e20ba28129589c5f05f9c8 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Fri, 9 Feb 2024 15:39:12 +0100
Subject: [PATCH] fix main
---
main.py | 22 +++++++++++-----------
1 file changed, 11 insertions(+), 11 deletions(-)
diff --git a/main.py b/main.py
index bf1509e..bff9cd6 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