From 51b744a26960a69d4e880c8c776fd3cb84776e06 Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Tue, 16 May 2023 08:42:10 +0200
Subject: [PATCH] Changing MUTAN rank to 10 instead of 1
---
train_rsvqa.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train_rsvqa.py b/train_rsvqa.py
index 6d1e448..0734109 100644
--- a/train_rsvqa.py
+++ b/train_rsvqa.py
@@ -240,7 +240,7 @@ def mutan(fusion_in: int, fusion_out: int):
'dim_mm': fusion_out,
'dropout_hv': 0.1,
'dropout_hq': 0.1,
- 'R': 1
+ 'R': 10
}
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
f = MutanFusion(visual_embedding=False,
--
GitLab