Keras Project Template

Inspired from https://github.com/Ahmkel/Keras-Project-Template

CycleGAN is implemented as a sample model.

Features

• tensorpack powered dataloader
• supports save/load models and optimizers
• supports multi-gpu training
• backups source code and config used to train
• stops training if encounter nan loss
• collages losses to a single graph on tensorboard
• telegram notification on train start/end

Requirements

python >= 3.6

Install required packages

pip install -r requirements.txt

Download dataset

sh ./download_dataset.sh horse2zebra

Setup config

edit configs/cyclegan.yml file

To train

python train.py --config configs/cyclegan.yml

Training from a certain checkpoint.

python train.py --config configs/cyclegan.yml --chkpt 123

Visualize via Tensorboard

tensorboard --logdir /path/to/experiment/dir/tensorboard/

Setup telegram notification

Add telegram token and chat_id to environment variables as TELEGRAM_TOKEN and TELEGRAM_CHAT_ID