fix

pliang279 · Oct 19, 2021 · b1b71f4 · b1b71f4
1 parent 09a2219
commit b1b71f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/multimedia/avmnist_simple_late_fusion.py b/examples/multimedia/avmnist_simple_late_fusion.py
@@ -15,7 +15,7 @@
 
 fusion=Concat().cuda()
 
-train(encoders,fusion,head,traindata,validdata,3,optimtype=torch.optim.SGD,lr=0.1,weight_decay=0.0001)
+train(encoders,fusion,head,traindata,validdata,30,optimtype=torch.optim.SGD,lr=0.1,weight_decay=0.0001)
 
 print("Testing:")
 model=torch.load('best.pt').cuda()