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

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

diff --git a/MOAFTrainDDP.py b/MOAFTrainDDP.py
index bf95fb8..c1d83ff 100644
--- a/MOAFTrainDDP.py
+++ b/MOAFTrainDDP.py
@@ -112,7 +112,7 @@ def fit(rank, world_size, cfg):
         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)