parser.add_argument('--mod',type=str,default='mod2',help='chooses which dataset model for test. mod1....')
parser.add_argument('--pretrained_model_path',type=str,default='./checkpoints/ANN_mod1_231224-222338/ANN_mod1_opt.pt',help='pretrained model file load path')