def parse(conf: Conf):
parser = argparse.ArgumentParser(description='Train img to video model')
parser = conf.add_default_args(parser)
parser.add_argument('teacher', type=str)
parser.add_argument('--teacher_chk_name', type=str, default='chk_end')
parser.add_argument('--student', type=str)
parser.add_argument('--student_chk_name', type=str, default='chk_end')
parser.add_argument('--exp_name', type=str, default=str(uuid4()))
parser.add_argument('--num_generations', type=int, default=1)
parser.add_argument('--eval_epoch_interval', type=int, default=50)
parser.add_argument('--print_epoch_interval', type=int, default=5)
parser.add_argument('--lr', type=float, default=1e-4)
parser.add_argument('--lr_decay', type=float, default=0.1)
parser.add_argument('--temp', type=float, default=10.)
parser.add_argument('--lambda_coeff', type=float, default=0.0001)
parser.add_argument('--kl_coeff', type=float, default=0.1)
parser.add_argument('--num_train_images', type=int, default=8)
parser.add_argument('--num_student_images', type=int, default=2)
parser.add_argument('--train_strategy', type=str, default='multiview',
choices=['multiview', 'temporal'])
parser.add_argument('--num_epochs', type=int, default=400)
parser.add_argument('--gamma', type=float, default=0.1)
parser.add_argument('--first_milestone', type=int, default=300)
parser.add_argument('--step_milestone', type=int, default=50)
parser.add_argument('--reinit_l4', type=str2bool, default=True)
parser.add_argument('--reinit_l3', type=str2bool, default=False)
parser.add_argument('--logits_dist', type=str, default='kl',
choices=['kl', 'mse'])
args = parser.parse_args()
args.use_random_erasing = True
return args
def parse(conf: Conf):
parser = argparse.ArgumentParser(description='Train img to video model')
parser = conf.add_default_args(parser)
parser.add_argument('trinet_folder',
type=str,
help='Path to TriNet base folder.')
parser.add_argument('--trinet_chk_name',
type=str,
help='checkpoint name',
default='chk_end')
args = parser.parse_args()
args.train_strategy = 'chunk'
args.use_random_erasing = False
args.num_train_images = 0
return args
def parse(conf: Conf):
parser = argparse.ArgumentParser(description='Train img to video model')
parser = conf.add_default_args(parser)
parser.add_argument('--exp_name',
type=str,
default=str(uuid4()),
help='Experiment name.')
parser.add_argument('--metric',
type=str,
default='euclidean',
choices=['euclidean', 'cosine'],
help='Metric for distances')
parser.add_argument('--num_train_images',
type=int,
default=8,
help='Num. of bag images.')
parser.add_argument('--num_epochs', type=int, default=300)
parser.add_argument('--eval_epoch_interval', type=int, default=50)
parser.add_argument('--save_epoch_interval', type=int, default=50)
parser.add_argument('--print_epoch_interval', type=int, default=5)
parser.add_argument('--wd', type=float, default=1e-5)
parser.add_argument('--gamma', type=float, default=0.1)
parser.add_argument('--first_milestone', type=int, default=200)
parser.add_argument('--step_milestone', type=int, default=50)
parser.add_argument('--use_random_erasing', type=str2bool, default=True)
parser.add_argument('--train_strategy',
type=str,
default='chunk',
choices=['multiview', 'chunk'])
args = parser.parse_args()
return args
def parse(conf: Conf):
parser = argparse.ArgumentParser(description='Train img to video model')
parser = conf.add_default_args(parser)
parser.add_argument('net1', type=str, help='Path to TriNet base folder.')
parser.add_argument('--chk_net1',
type=str,
help='checkpoint name',
default='chk_end')
parser.add_argument('net2', type=str, help='Path to TriNet base folder.')
parser.add_argument('--chk_net2',
type=str,
help='checkpoint name',
default='chk_end')
parser.add_argument('--dest_path', type=Path, default='/tmp/heatmaps_out')
args = parser.parse_args()
args.train_strategy = 'multiview'
args.use_random_erasing = False
args.num_train_images = 0
args.img_test_batch = 32
return args