()
| 38 | |
| 39 | # Command line arguments setup |
| 40 | def parse_args(): |
| 41 | parser = argparse.ArgumentParser(description='Modify model to use custom MyDense layer.', |
| 42 | epilog='Usage: python recomplier.py --model_file YOUR_MODEL_FILE') |
| 43 | parser.add_argument('--model', help="File containing the model (e.g., model.py)", default="", type=str) |
| 44 | return parser.parse_args() |
| 45 | |
| 46 | def main(): |
| 47 | args = parse_args() |