🐛 (classifier): Fix missing Namespace entry

token-tricksters · Jul 5, 2023 · 23b46c8 · 23b46c8
1 parent bd46581
commit 23b46c8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/classifier.py b/classifier.py
@@ -454,7 +454,8 @@ def get_args():
         option=args.option,
         dev_out='predictions/' + args.option + '-sst-dev-out.csv',
         test_out='predictions/' + args.option + '-sst-test-out.csv',
-        logdir=args.logdir
+        logdir=args.logdir,
+        optimizer=args.optimizer
     )
 
     train(config)