def make_options_from_flags(FLAGS):
if FLAGS.json_config is not None:
options = read_options_from_file(FLAGS.json_config)
else:
options = Options() # the default value stored in config.Options
if FLAGS.shuffle is not None:
options.shuffle = FLAGS.shuffle
if FLAGS.net_mode is not None:
options.net_mode = FLAGS.net_mode
if FLAGS.data_mode is not None:
options.data_mode = FLAGS.data_mode
if FLAGS.load_mode is not None:
options.load_mode = FLAGS.load_mode
if FLAGS.fix_level is not None:
options.fix_level = FLAGS.fix_level
if FLAGS.init_learning_rate is not None:
options.base_lr = FLAGS.init_learning_rate
if FLAGS.optimizer != 'sgd':
options.optimizer = FLAGS.optimizer
if FLAGS.weight_decay != 0.00004:
options.weight_decay = FLAGS.weight_decay
if FLAGS.global_label is not None:
options.data_mode == 'global_label'
options.global_label = FLAGS.global_label
if options.load_mode != 'normal':
if FLAGS.backbone_model_path is not None:
options.backbone_model_path = FLAGS.backbone_model_path
else:
options.backbone_model_path = None
return options
# model_folder = home_dir+'data/mask_imagenet_solid_rd/0_checkpoint/'
model_path = None
try:
model_path = get_last_checkpoint_in_folder(model_folder)
except:
pass
# model_path = '/home/tdteach/data/mask_test_gtsrb_f1_t0_c11c12_solid/_checkpoint/model.ckpt-3073'
# model_path = '/home/tdteach/data/mask_test_gtsrb_f1_t0_nc_solid/_checkpoint/model.ckpt-27578'
# model_path = '/home/tdteach/data/_checkpoint/model.ckpt-0'
# model_path = home_dir+'data/cifar10_models/benign_all'
# subname = 'strip'
# model_path = home_dir+'data/gtsrb_models/benign_all'
# model_path = home_dir+'data/gtsrb_models/f1t0c11c12'
# model_path = home_dir+'data/imagenet_models/f2t1c11c12'
# model_path = home_dir+'data/imagenet_models/benign_all'
options.backbone_model_path = model_path
options.net_mode = 'normal'
# options.load_mode = 'bottom_affine'
options.load_mode = 'all'
options.num_epochs = 60
# options.data_mode = 'poison'
options.data_mode = 'normal'
#label_list = list(range(20))
