fixing warmup and some messages

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

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 ===")