Train model python train.py --gpus 0 --dataset nyc python train.py --gpus 0 --dataset chicago Requirements tensorflow 2.3.0 Keras 2.1.3