shalpin87
/

diffusion_conditional

DDPMConditionalPipeline

Model card Files Files and versions

Sean Halpin commited on Aug 16, 2022

Commit

2cb7709

·

1 Parent(s): 64edef4

update model card README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ on the `CelebA` dataset.
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
 - train_batch_size: 32
-- eval_batch_size: 32
 - gradient_accumulation_steps: 1
 - optimizer: AdamW with betas=(0.95, 0.999), weight_decay=1e-06 and epsilon=1e-08
 - lr_scheduler: cosine

 The following hyperparameters were used during training:
 - learning_rate: 0.0001
 - train_batch_size: 32
+- eval_batch_size: 16
 - gradient_accumulation_steps: 1
 - optimizer: AdamW with betas=(0.95, 0.999), weight_decay=1e-06 and epsilon=1e-08
 - lr_scheduler: cosine