From b1d5a55816b953ce88f4796cd1082bbf815fcd91 Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Mon, 15 May 2023 08:47:40 +0200
Subject: [PATCH] switching from precision in trainer to matmul precision in
torch
---
train_rsvqa.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 4461057..6d1e448 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -269,6 +269,8 @@ def main(
pl.seed_everything(seed, workers=True)
+ torch.set_float32_matmul_precision("medium")
+
img_size = 120
channels = 10
@@ -326,7 +328,6 @@ def main(
logger=wandb_logger,
check_val_every_n_epoch=2,
callbacks=[checkpoint_callback, early_stopping_callback, lr_monitor],
- precision="16",
)
model = LitVisionEncoder(config=model_config, lr=lr)
--
GitLab