kaiza_hikaru/MOAF
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix mmlp mode

Browse Source
This commit is contained in:
kaiza_hikaru 2025-11-02 15:15:15 +00:00
parent a8969ea76e
commit bdccc22b9f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
MOAFTrainDDP.py
View File

@ -112,7 +112,7 @@ def fit(rank, world_size, cfg):
fusion_depth_list = [int(ch) for ch in model_type[2:]] fusion_depth_list = [int(ch) for ch in model_type[2:]]
model = MOAFWithSE(fusion_depth_list).to(device) model = MOAFWithSE(fusion_depth_list).to(device)
elif "mmlp" in model_type: elif "mmlp" in model_type:
model = MOAFWithMMLP(fusion_depth_list).to(device) model = MOAFWithMMLP().to(device)
else: else:
model = MOAFNoFusion().to(device) model = MOAFNoFusion().to(device)