| import argparse | |
| import time | |
| def train(epochs): | |
| for epoch in range(epochs): | |
| print("Epoch %d/%d training..." % (epoch+1, epochs)) | |
| time.sleep(2) | |
| print("Epoch %d complete, accuracy: %.2f" % (epoch+1, 0.8 + epoch*0.01)) | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--epochs", type=int, default=1) | |
| args = parser.parse_args() | |
| train(args.epochs) | |