From 955083968e153ca18e9c5019772a23f4597a942b Mon Sep 17 00:00:00 2001
From: kaiza_hikaru <kaiza_hikaru@163.com>
Date: Sun, 2 Nov 2025 15:15:46 +0000
Subject: [PATCH] fix mmlp mode 2

---
 MOAFTrain.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MOAFTrain.py b/MOAFTrain.py
index 8dde796..9c3d7de 100644
--- a/MOAFTrain.py
+++ b/MOAFTrain.py
@@ -143,7 +143,7 @@ def main():
         fusion_depth_list = [int(ch) for ch in model_type[2:]]
         model = MOAFWithSE(fusion_depth_list).to(device)
     elif "mmlp" in model_type:
-        model = MOAFWithMMLP(fusion_depth_list).to(device)
+        model = MOAFWithMMLP().to(device)
     else:
         model = MOAFNoFusion().to(device)
     print_with_timestamp("Model Loaded")