close
常見輸入引數解析三步驟
1. parser
2. command line parsing: args
3. call main function with args
import argparse
# 1. parser
parser = argparse.ArgumentParser()
parser.add_argument("--window_size", type=int, default=4)
parser.add_argument("--model_path", type=str,default="./models/best.pth")
parser.add_argument("--video_paths", type=str,default=video_paths)
parser.add_argument("--labels", type=str, default=labels)
parser.add_argument("--num_epochs", type=int, default=130)
parser.add_argument("--loss_target", type=float, default=0.00001)
# 2. command line command parsing method
args = parser.parse_args()
# 3. Call your main() function here
main(args.window_size, args.model_path, video_paths, labels, args.num_epochs, args.loss_target)
全站熱搜
留言列表
