Spaces:
Running
on
A10G
Running
on
A10G
File size: 2,068 Bytes
bcec54e |
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 47 48 49 50 51 |
# ------------------------------------------------------------------------------
# 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
|