-
Notifications
You must be signed in to change notification settings - Fork 4
/
parser.py
50 lines (29 loc) · 2.58 KB
/
parser.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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import argparse
parser = argparse.ArgumentParser(description='Arguments for running the scripts')
#ARGS FOR LOADING THE DATASET
parser.add_argument('--data_dir',type=str,default='../datasets/',help='path to the unziped dataset directories(H36m)')
parser.add_argument('--input_n',type=int,default=10,help="number of model's input frames")
parser.add_argument('--output_n',type=int,default=10,help="number of model's output frames")
parser.add_argument('--test_output_n',type=int,default=10,help="number of model's test frames")
parser.add_argument('--n_pre',type=int,default=35,help="number of dct coefficients")
parser.add_argument('--skip_rate',type=int,default=2,help='rate of frames to skip,defaults=2 for H36M')
parser.add_argument('--global_translation', action='store_true',help='predicting global translation for H36M')
#ARGS FOR THE MODEL
parser.add_argument('--st_gcnn_dropout',type=float,default=.1,help= 'st-gcnn dropout')
#ARGS FOR THE TRAINING
parser.add_argument('--mode',type=str,default='train',choices=['train','test','viz'],help= 'Choose to train,test or visualize from the model.Either train,test or viz')
parser.add_argument('--n_epochs',type=int,default=50,help= 'number of epochs to train')
parser.add_argument('--batch_size',type=int,default=256,help= 'batch size')
parser.add_argument('--batch_size_test',type=int,default=256,help= 'batch size for the test set')
parser.add_argument('--lr',type=int,default=1e-02,help= 'Learning rate of the optimizer')
parser.add_argument('--use_scheduler',type=bool,default=True,help= 'use MultiStepLR scheduler')
parser.add_argument('--milestones',type=list,default=[15,25,35,40],help= 'the epochs after which the learning rate is adjusted by gamma')
parser.add_argument('--gamma',type=float,default=0.1,help= 'gamma correction to the learning rate, after reaching the milestone epochs')
parser.add_argument('--clip_grad',type=float,default=None,help= 'select max norm to clip gradients')
parser.add_argument('--model_path',type=str,default='./checkpoints/CKPT_3D_H36M',help= 'directory with the models checkpoints ')
parser.add_argument('--version',type=str,default='long',help= 'model version (long or short)')
#FLAGS FOR THE VISUALIZATION
parser.add_argument('--visualize_from',type=str,default='test',choices =['train','val','test'],help= 'choose data split to visualize from(train-val-test)')
parser.add_argument('--actions_to_consider',default='all',help= 'Actions to visualize.Choose either all or a list of actions')
parser.add_argument('--n_viz',type=int,default='2',help= 'Numbers of sequences to visaluze for each action')
args = parser.parse_args()