fix train

run_interaction.py

@@ -176,7 +176,7 @@ if __name__ == '__main__':
                         help='number of data loading workers (default: 0)')
     parser.add_argument('--epochs', default=50, type=int, metavar='N',
                         help='number of total epochs to run')
-    parser.add_argument('--task', choices=['train', 'test', 'channel', 'ER', 'GPCR', 'kinase'],
+    parser.add_argument('--task', choices=['train', 'train_z_10', 'train_z_100', 'test', 'channel', 'ER', 'GPCR', 'kinase'],
                         default='train', type=str, metavar='TASK',
                         help='Task name. Could be train, test, channel, ER, GPCR, kinase.')
     parser.add_argument('--lr', '--learning-rate', default=1e-5, type=float,
@@ -194,11 +194,11 @@ if __name__ == '__main__':
 
     # local test
 
-    args.task = 'train_z_10'
-    args.savedir = 'local_test_train'
-    args.epochs = 10
-    args.lr = 1e-5
-    args.config = './config/config_layer_3.json'
+    # args.task = 'train_z_10'
+    # args.savedir = 'local_test_train'
+    # args.epochs = 10
+    # args.lr = 1e-5
+    # args.config = './config/config_layer_3.json'