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

fix test

Browse Source
This commit is contained in:
kaiza_hikaru 2025-11-03 10:33:16 +08:00
parent fd7a9dba2e
commit 3d48dbdd17
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
MOAFTest.py
View File

@ -73,7 +73,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)