train.py

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)