From 8cc1539ee5e0f49d3bcd793184e3df245e990293 Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Mon, 15 May 2023 08:44:48 +0200
Subject: [PATCH] amp flag seems to have been invalid
---
train_rsvqa.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 47ac1b2..4461057 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",
+ precision="16",
)
model = LitVisionEncoder(config=model_config, lr=lr)
--
GitLab