def get_lucid_stimuli_compute_parser():
parser = argparse.ArgumentParser(
description='Compute the optimal stimuli for raw units using lucid')
parser.add_argument('--gpu', default='0', type=str, action='store')
parser = bs_fit.add_load_settings(parser)
parser = bs_fit.add_model_settings(parser)
return parser
def get_stimuli_compute_parser():
parser = argparse.ArgumentParser(
description='The script to compute the optimal stimuli')
parser = add_stimuli_settings(parser)
parser = bs_fit.add_load_settings(parser)
parser = bs_fit.add_model_settings(parser)
return parser
def get_circular_var_parser():
parser = argparse.ArgumentParser(
description='The script to compute circular variance')
parser.add_argument(
'--gpu', default='0', type=str, action='store')
parser.add_argument(
'--id_suffix', default='saver', type=str, action='store')
parser = bs_fit.add_load_settings(parser)
parser = bs_fit.add_model_settings(parser)
return parser
def get_compute_raw_response_parser():
parser = argparse.ArgumentParser(
description='Compute the response for inputs')
parser.add_argument('--gpu', default='0', type=str, action='store')
parser.add_argument('--file_path', type=str, action='store', required=True)
parser.add_argument('--layer', type=str, action='store', required=True)
parser.add_argument('--batch_size',
default=64,
type=int,
action='store',
help='Batch size')
parser = bs_fit.add_load_settings(parser)
parser = bs_fit.add_model_settings(parser)
return parser
def get_raw_pt_stimuli_compute_parser():
parser = argparse.ArgumentParser(
description='Compute the optimal stimuli for pytorch models')
parser.add_argument('--gpu', default='0', type=str, action='store')
parser.add_argument('--tv_wd', default=1e-4, type=float, action='store')
parser.add_argument('--wd', default=1e-4, type=float, action='store')
parser.add_argument('--lr', default=0.05, type=float, action='store')
parser.add_argument('--global_start_idx',
default=None,
type=int,
action='store')
parser.add_argument('--num_batches', default=4, type=int, action='store')
parser.add_argument('--layer_start_idx',
default=None,
type=int,
action='store')
parser.add_argument('--layer_len_idx', default=1, type=int, action='store')
parser = bs_fit.add_load_settings(parser)
parser = bs_fit.add_model_settings(parser)
return parser