def main():
parser = get_parser()
args = parser.parse_args()
os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
train_framework = TrainFramework(args)
train_framework.train()
def load_prev_func(args, load_func):
train_args = cmd_parser.get_parser().parse_args([])
train_args.load_setting_func = load_func
train_args = cmd_parser.load_setting(train_args)
args.loadport = train_args.nport
args.load_dbname = train_args.dbname
args.load_collname = train_args.collname
args.loadexpId = train_args.expId
return args, train_args
def rp_s0(args):
args = transfer_basic(args)
train_args = cmd_parser.get_parser().parse_args([])
train_args.pathconfig = 'rp_resnet18.cfg'
args.network_func = lambda: trans_network_cfg(train_args)
args = clear_load_params(args)
step_num = 1181162
args.ckpt_file = '/mnt/fs4/chengxuz/tpu_ckpts/rp_res18/model.ckpt-%i' % step_num
args.ignorebname_new = 0
args.add_batchname = '_rp'
args.rp_dp_tl = 1
args.no_prep = 1
args.expId = 'rp_s0'
args.init_lr = 0.01
return args
def depth_network_cfg_setting(args):
train_args = cmd_parser.get_parser().parse_args([])
train_args.pathconfig = 'depth_resnet18_up4.cfg'
args.network_func = lambda: trans_network_cfg(train_args)
return args