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

fix mmlp mode 2

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

2
MOAFTrain.py
View File

@ -143,7 +143,7 @@ def main():
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)
print_with_timestamp("Model Loaded") print_with_timestamp("Model Loaded")