Spaces:
Running
on
A10G
Running
on
A10G
| # ------------------------------------------------------------------------------ | |
| # The code is from GLPDepth (https://github.com/vinvino02/GLPDepth). | |
| # For non-commercial purpose only (research, evaluation etc). | |
| # Modified by Zigang Geng (zigang@mail.ustc.edu.cn). | |
| # ------------------------------------------------------------------------------ | |
| from configs.base_options import BaseOptions | |
| import argparse | |
| def str2bool(v): | |
| if isinstance(v, bool): | |
| return v | |
| if v.lower() in ('yes', 'true', 't', 'y', '1'): | |
| return True | |
| elif v.lower() in ('no', 'false', 'f', 'n', '0'): | |
| return False | |
| else: | |
| raise argparse.ArgumentTypeError('Boolean value expected.') | |
| class TrainOptions(BaseOptions): | |
| def initialize(self): | |
| parser = BaseOptions.initialize(self) | |
| # experiment configs | |
| parser.add_argument('--epochs', type=int, default=25) | |
| parser.add_argument('--max_lr', type=float, default=5e-4) | |
| parser.add_argument('--min_lr', type=float, default=3e-5) | |
| parser.add_argument('--weight_decay', type=float, default=5e-2) | |
| parser.add_argument('--layer_decay', type=float, default=0.9) | |
| parser.add_argument('--crop_h', type=int, default=448) | |
| parser.add_argument('--crop_w', type=int, default=576) | |
| parser.add_argument('--log_dir', type=str, default='./logs') | |
| # logging options | |
| parser.add_argument('--val_freq', type=int, default=1) | |
| parser.add_argument('--pro_bar', type=str2bool, default='False') | |
| parser.add_argument('--save_freq', type=int, default=1) | |
| parser.add_argument('--print_freq', type=int, default=100) | |
| parser.add_argument('--save_model', action='store_true') | |
| parser.add_argument( | |
| '--resume-from', help='the checkpoint file to resume from') | |
| parser.add_argument('--auto_resume', action='store_true') | |
| parser.add_argument('--save_result', action='store_true') | |
| return parser | |