Skip to content

Ouwen/keras-scaffold

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages