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