Train the model python train.py --train_dir=ckpt/face --batch_size=4 --initial_learning_rate=0.0001 --dataset_dir=<your-path> Run the demo Demo