diff --git a/ben_publication/BENv2ImageClassifier.py b/ben_publication/BENv2ImageClassifier.py
index 07392307193081c761e95f648fa0f87f5c460ee6..77736ad33ad7ec3080148beb8c82185261165ddd 100644
--- a/ben_publication/BENv2ImageClassifier.py
+++ b/ben_publication/BENv2ImageClassifier.py
@@ -35,7 +35,7 @@ class BENv2ImageEncoder(pl.LightningModule, PyTorchModelHubMixin):
     ):
         super().__init__()
         self.lr = lr
-        self.warmup = warmup
+        self.warmup = None if warmup < 0 else warmup
         self.config = config
         assert config.network_type == ILMType.IMAGE_CLASSIFICATION
         assert config.classes == 19
diff --git a/scripts/improve_hf_training.py b/scripts/improve_hf_training.py
index 2e55d682e64895670d46ce6ef40e0209580c0449..612ae64b1e05689b7681ed461c24fa45f516c3a6 100644
--- a/scripts/improve_hf_training.py
+++ b/scripts/improve_hf_training.py
@@ -188,8 +188,10 @@ def main(
             print(f"Pushing to {push_path}")
             model.push_to_hub(push_path, commit_message=f"Upload {model_name}")
             print("=== Done ===")
+        else:
+            print("=== Skipping upload to Huggingface Hub because no entity was provided ===")
     else:
-        print("=== Skipping upload to Huggingface Hub ===")
+        print("=== Skipping upload to Huggingface Hub because the new model did not improve the compare metric ===")
 
     print("=== Training finished ===")