Set Mamba batch_size=32 to avoid OOM

params_storage/params_mamba_single_gpu.py

@@ -30,7 +30,7 @@ opt_exp.data = "rw_to_rw"           # Same Jacobs Hall dataset as Fig 10b
 opt_exp.n_decoders = 1               # Single decoder (location only)
 opt_exp.model_type = "mamba"         # Selects Mamba pipeline in train_and_test.py
 
-opt_exp.batch_size = 64
+opt_exp.batch_size = 32              # 64 OOMs on H100 with mamba-ssm
 opt_exp.ds_step_trn = 1
 opt_exp.ds_step_tst = 1
 opt_exp.weight_decay = 1e-5