main
import argparse def get_args(): parser = argparse.ArgumentParser() parser.add_argument("--max_epochs", type=int, default=5) return parser.parse_args()