change train

run_interaction.py

@@ -159,7 +159,7 @@ def main(args):
     print('model name : BertAffinity')
     print('task name : {}'.format(args.task))
 
-    if args.task in ['train', 'train_z_10', 'train_z_100']:
+    if args.task in ['train', 'train_z_1', 'train_z_10', 'train_z_100']:
         train(args, model, dataset)
 
     elif args.task in ['test']: