| 142 | |
| 143 | |
| 144 | def parse_config_args(exp_name): |
| 145 | parser = argparse.ArgumentParser(description=exp_name) |
| 146 | parser.add_argument('--cfg', type=str, |
| 147 | default='../experiments/workspace/retrain/retrain.yaml', |
| 148 | help='configuration of cream') |
| 149 | parser.add_argument('--local_rank', type=int, default=0, |
| 150 | help='local_rank') |
| 151 | args = parser.parse_args() |
| 152 | |
| 153 | cfg.merge_from_file(args.cfg) |
| 154 | converted_cfg = convert_lowercase(cfg) |
| 155 | |
| 156 | return args, converted_cfg |
| 157 | |
| 158 | |
| 159 | def get_model_flops_params(model, input_size=(1, 3, 224, 224)): |