Tensorflow 2 Project Template
Inspired from https://github.com/Ahmkel/Keras-Project-Template
Pix2Pix is implemented as a sample model.
Features
- supports multi-gpu training
- supports save model and load model in order to continue to train the model
- backups source code used to train
How to train
Requirements
python >= 3.7
Install required packages
pip install -r requirements.txt
Download pix2pix datasets
./datasets/download_pix2pix_dataset.sh facades
Setup the config file
edit configs/pix2pix.yml file
Train
python train.py --config configs/pix2pix.yml
Train from a certain checkpoint
python train.py --config configs/pix2pix.yml --checkpoint 100
Run Tensorboard
tensorboard --logdir /path/to/experiment/dir/tensorboard/