The following repo is a starting scaffold for modeling projects.
- CycleGAN
- Pix2Pix
- Unet Attention Block
- Residual Block
- TODO: focal loss
- TODO: diversity sensitive GAN
log_code.py
, logs code to GCP bucket before trainingcopy_keras_model.py
, copies keras models to GCP bucketsave_multi_model.py
, saves multiple keras models each epoch
generate_images.py
, writes images to tensorboardget_csv_metrics.py
, writes metrics of interest to a csv file.
- GCP hyperparam tuning
- TODO: Axios