python train.py --gpus 0 --dataset nyc

python train.py --gpus 0 --dataset chicago

Requirements

tensorflow 2.3.0

Keras 2.1.3