Skip to content

A project template to start writing deep learning models using Tensorflow 2.

Notifications You must be signed in to change notification settings

luke-han/tf2-project-template

Repository files navigation

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/

About

A project template to start writing deep learning models using Tensorflow 2.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published