diff --git a/train_lit4rsvqa.py b/train_lit4rsvqa.py
index 6117e57893f592834c848211c48e008905604695..4189a697f996373b84d6bcaa216dce0a42048d46 100644
--- a/train_lit4rsvqa.py
+++ b/train_lit4rsvqa.py
@@ -290,7 +290,7 @@ def main(
     monitor_str = "AA"
     # checkpointing
     checkpoint_callback = ModelCheckpoint(
-        monitor="val/f1",
+        monitor=monitor,
         dirpath="./checkpoints",
         filename=f"{wandb_logger.experiment.name}-seed=" +
                  str(seed) + "-epoch={epoch:03d}-" + f"{monitor_str}" + "={" +
@@ -300,8 +300,6 @@ def main(
         mode="max",
         save_last=True
     )
-    early_stopping_callback = EarlyStopping(monitor=monitor, min_delta=0.00,
-                                            patience=25, verbose=False, mode="max")
     lr_monitor = LearningRateMonitor(logging_interval='step')
 
     trainer = pl.Trainer(
@@ -310,7 +308,7 @@ def main(
         log_every_n_steps=5,
         logger=wandb_logger,
         check_val_every_n_epoch=2,
-        callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
+        callbacks=[checkpoint_callback, lr_monitor],
 
     )