Skip to content
Snippets Groups Projects
Commit 51b744a2 authored by Leonard Wayne Hackel's avatar Leonard Wayne Hackel
Browse files

Changing MUTAN rank to 10 instead of 1

parent b1d5a558
No related branches found
No related tags found
Loading
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment