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..')
