change task

run_interaction.py

@@ -137,9 +137,8 @@ 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', 'train_z_1', 'train_z_10', 'train_z_100', 'test', 'channel', 'ER', 'GPCR', 'kinase', 'train_mol'],
-                        default='train', type=str, metavar='TASK',
-                        help='Task name. Could be train, test, channel, ER, GPCR, kinase.')
+    parser.add_argument('--task', default='train', type=str, metavar='TASK',
+                        help='Task name. Could be train, test, channel, ER, GPCR, kinase or else.')
     parser.add_argument('--lr', '--learning-rate', default=1e-5, type=float,
                         metavar='LR', help='initial learning rate', dest='lr')
     parser.add_argument('--config', default='./config/config.json', type=str, help='model config file path')