-
Notifications
You must be signed in to change notification settings - Fork 1
/
batch_main.py
25 lines (23 loc) · 1.01 KB
/
batch_main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from main import process_main_args
from misc.datasets import datasets
import argparse
if __name__ == "__main__":
""" Enable the run of multiple target subjects one after the other using the main functions """
parser = argparse.ArgumentParser()
parser.add_argument("--tl_mode", type=str)
parser.add_argument("--source_dataset", type=str)
parser.add_argument("--target_dataset", type=str)
parser.add_argument("--target_subject",type=str)
parser.add_argument("--model", type=str)
parser.add_argument("--params", type=str)
parser.add_argument("--params_ft", type=str)
parser.add_argument("--weights", type=str)
parser.add_argument("--eval_mode", type=str)
parser.add_argument("--split", type=int)
parser.add_argument("--log", type=str)
parser.add_argument("--exp", type=str)
parser.add_argument("--plot", type=bool)
args = parser.parse_args()
for subject in datasets[args.target_dataset]["subjects"]:
args.target_subject = subject
process_main_args(args)