tensorflow 2.0 example Requirements tensorflow-datasets==1.0.1 tensorflow==2.0.0-alpha0 Training $ python main.py $ python main.py --trainer keras