Merge branch 'main' of https://github.com/pliang279/MultiBench into main

pliang279 · Oct 21, 2021 · ed01592 · ed01592
2 parents cd484d7 + a66a74f
commit ed01592
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training_structures/unimodal.py b/training_structures/unimodal.py
@@ -158,7 +158,7 @@ def test(encoder, head, test_dataloaders_all, dataset='default', method_name='My
  if no_robust:
  def testprocess():
  single_test(encoder,head,test_dataloaders_all,auprc,modalnum,task,criterion)
- all_in_one_test(testprocess,[model])
+ all_in_one_test(testprocess,[encoder,head])
  return
  def testprocess():
  single_test(encoder, head, test_dataloaders_all[list(test_dataloaders_all.keys())[0]][0], auprc, modalnum, task, criterion)