/keras-scaffold

Primary LanguagePython

Keras Scaffold

The following repo is a starting scaffold for modeling projects.

Implemented Models

  • CycleGAN
  • Pix2Pix

Implemented Blocks

  • Unet Attention Block
  • Residual Block

Regularizations

  • TODO: focal loss
  • TODO: diversity sensitive GAN

Implemented Callbacks

  • log_code.py, logs code to GCP bucket before training
  • copy_keras_model.py, copies keras models to GCP bucket
  • save_multi_model.py, saves multiple keras models each epoch

Custom Callbacks

  • generate_images.py, writes images to tensorboard
  • get_csv_metrics.py, writes metrics of interest to a csv file.

Hyperparam tuning

  • GCP hyperparam tuning
  • TODO: Axios