From 974eb50eb425cfdb06d4f51fcfcccd701223f196 Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Wed, 3 May 2023 08:29:51 +0200
Subject: [PATCH] adding vision pretraining tag automatically on train start
---
train_lit4rsvqa.py | 2 ++
train_rsvqa.py | 2 ++
2 files changed, 4 insertions(+)
diff --git a/train_lit4rsvqa.py b/train_lit4rsvqa.py
index 04975c2..bce447f 100644
--- a/train_lit4rsvqa.py
+++ b/train_lit4rsvqa.py
@@ -269,6 +269,8 @@ def main(
tags = ["Training", vision_model, text_model]
if test_run:
tags += ["Test Run"]
+ if vision_checkpoint is not None:
+ tags += ["Vision Pretraining"]
wandb_logger = WandbLogger(project=f"LiT4RSVQA",
log_model=True,
tags=tags, # keyword arg directly to wandb.init()
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 48e92c1..e34232f 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -292,6 +292,8 @@ def main(
tags = ["Training", vision_model, text_model]
if test_run:
tags += ["Test Run"]
+ if vision_checkpoint is not None:
+ tags += ["Vision Pretraining"]
wandb_logger = WandbLogger(project=f"LiT4RSVQA",
log_model=True,
tags=tags, # keyword arg directly to wandb.init()
--
GitLab