From c3fb68c44452f20056082e9f0ca01ed9b206b60a Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Wed, 10 May 2023 08:49:24 +0200
Subject: [PATCH] changing default values for epochs and val epochs
---
train_lit4rsvqa.py | 4 ++--
train_rsvqa.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/train_lit4rsvqa.py b/train_lit4rsvqa.py
index 3076169..09ef31b 100644
--- a/train_lit4rsvqa.py
+++ b/train_lit4rsvqa.py
@@ -243,7 +243,7 @@ def main(
vision_model: str = "mobilevit_s",
text_model: str = "prajjwal1/bert-tiny",
lr: float = 1e-3,
- epochs: int = 100,
+ epochs: int = 10,
batch_size: int = 32,
seed: int = 42,
data_dir: str = None,
@@ -307,7 +307,7 @@ def main(
accelerator="auto",
log_every_n_steps=5,
logger=wandb_logger,
- check_val_every_n_epoch=5,
+ check_val_every_n_epoch=2,
callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 1e2beab..82b18c8 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -253,7 +253,7 @@ def main(
vision_model: str = "resnet152",
text_model: str = "bert-base-uncased",
lr: float = 1e-3,
- epochs: int = 100,
+ epochs: int = 10,
batch_size: int = 32,
seed: int = 42,
data_dir: str = None,
@@ -324,7 +324,7 @@ def main(
accelerator="auto",
log_every_n_steps=5,
logger=wandb_logger,
- check_val_every_n_epoch=5,
+ check_val_every_n_epoch=2,
callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
)
--
GitLab