updata

run_interaction.py

@@ -132,7 +132,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', 'train_z_1', 'train_z_10', 'train_z_100', 'test', 'channel', 'ER', 'GPCR', 'kinase'],
+    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('--lr', '--learning-rate', default=1e-5, type=float,