Trained model v1

This release has a trained model which was trained for 50 epochs with following configuration

• image_size = 128  # the generated image resolution
  

• train_batch_size = 16
  

• eval_batch_size = 16  # how many images to sample during evaluation
  

• num_epochs = 50
  

• gradient_accumulation_steps = 1
  

• learning_rate = 1e-4
  

• lr_warmup_steps = 500
  

• save_image_epochs = 10
  

• save_model_epochs = 30
  

• mixed_precision = 'fp16'  # `no` for float32, `fp16` for automatic mixed precision
  

• output_dir = 'flowers-102-categories'  # the model namy locally and on the HF Hub
  

• push_to_hub = False  # whether to upload the saved model to the HF Hub
  

• hub_private_repo = False
  

• overwrite_output_dir = True  # overwrite the old model when re-running the notebook
  

• seed = 0