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