parser.add_argument("-train_epoch", type=int, default=30, help="No thread to load batch") parser.add_argument("-no_games_to_load", type=int, default=float("inf"), help="No games to use during training Default : all") parser.add_argument("-load_new", type=lambda x: bool(strtobool(x)), default="True", help="Start from checkpoint?") args = parser.parse_args() config, xp_manager = load_config(args) logger = logging.getLogger() # Load config batch_size = config['optimizer']['batch_size'] no_epoch = args.train_epoch if args.load_new and config['model']['image']['image_input'] == "rcnn": rcnn = True print("rcnn!") else: rcnn = False ############################### # LOAD DATA #############################
help="Load model parameters from specified checkpoint") parser.add_argument("-continue_exp", type=lambda x: bool(strtobool(x)), default="False", help="Continue previously started experiment?") parser.add_argument("-gpu_ratio", type=float, default=1., help="How many GPU ram is required? (ratio)") parser.add_argument("-no_thread", type=int, default=1, help="No thread to load batch") args = parser.parse_args() config, exp_identifier, save_path = load_config(args.config, args.exp_dir) logger = logging.getLogger() ############################### # LOAD DATA ############################# image_builder, crop_builder = None, None # Load image logger.info('Loading images..') use_resnet = False if 'image' in config['model']: logger.info('Loading images..') image_builder = get_img_builder(config['model']['image'], args.img_dir) use_resnet = image_builder.is_raw_image()
parser.add_argument("-img_dir", type=str, help='Directory with images to feed networks') parser.add_argument("-img_raw_dir", type=str, help='Directory with images to display') parser.add_argument("-crop_dir", type=str, help='Directory with crops') parser.add_argument("-exp_dir", type=str, required=False, help="Directory to output dialogue") parser.add_argument("-config", type=str, default="config/looper/config.uaqrah8g.json", help='Config file') parser.add_argument("-dict_file", type=str, default="dict.json", help="Dictionary file name") parser.add_argument("-networks_dir", type=str, help="Directory with pretrained networks") parser.add_argument("-qgen_identifier", type=str, default="3dc053450598749026e2ce6119e47d48_v1", required=False) parser.add_argument("-guesser_identifier", type=str, default='c48036b430ebca1c44a25188edb05034') parser.add_argument("-gpu_ratio", type=float, default=0.95, help="How many GPU ram is required? (ratio)") args = parser.parse_args() eval_config, exp_identifier = load_config(args, out_dir='./out/') # Load all networks configs logger = logging.getLogger() ############################### # LOAD DATA ############################# # Load image logger.info('Loading images..') image_builder = _create_image_builder_rcnn() crop_builder = None # Load data logger.info('Loading data..')