From 18fe808573b58b4b10ac65214677d9de4e8d95a3 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 21 May 2025 13:09:18 +0200
Subject: [PATCH] remove : early stopping
---
barlow_twin_like/main.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py
index 353a015..fef6992 100644
--- a/barlow_twin_like/main.py
+++ b/barlow_twin_like/main.py
@@ -220,11 +220,11 @@ def run():
_ = train_representation(model, data_train, optimizer, e, args.wandb)
if e % args.eval_inter == 0:
loss = test_representation(model, data_val, e, args.wandb)
- if loss < best_loss:
- save_model(model, args.save_path)
- best_loss = loss
-
- model.load_state_dict((torch.load(args.save_path, weights_only=True))) #load best model
+ # if loss < best_loss:
+ # save_model(model, args.save_path)
+ # best_loss = loss
+ #
+ # model.load_state_dict((torch.load(args.save_path, weights_only=True))) #load best model
for param in model.parameters(): # freezing representations before classifier training
param.requires_grad = False
--
GitLab