From 94c3bdc26f6c1bc83d397fdf9bc4a2b2c9e2bd4a Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Mon, 15 May 2023 08:42:47 +0200
Subject: [PATCH] adding mixed precision to rsvqa-t
---
train_rsvqa.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 05e2907..47ac1b2 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -326,7 +326,7 @@ def main(
logger=wandb_logger,
check_val_every_n_epoch=2,
callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
-
+ precision="16-mixed",
)
model = LitVisionEncoder(config=model_config, lr=lr)
--
GitLab