Hugging Face's logo

shalpin87
/
diffusion_conditional

Diffusers
English
DDPMConditionalPipeline
Model card Files
xet
 Community
1
Sean Halpin commited on
Commit
2cb7709
·
1 Parent(s): 64edef4

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ on the `CelebA` dataset.
38
  The following hyperparameters were used during training:
39
  - learning_rate: 0.0001
40
  - train_batch_size: 32
41
- - eval_batch_size: 32
42
  - gradient_accumulation_steps: 1
43
  - optimizer: AdamW with betas=(0.95, 0.999), weight_decay=1e-06 and epsilon=1e-08
44
  - lr_scheduler: cosine
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 0.0001
40
  - train_batch_size: 32
41
+ - eval_batch_size: 16
42
  - gradient_accumulation_steps: 1
43
  - optimizer: AdamW with betas=(0.95, 0.999), weight_decay=1e-06 and epsilon=1e-08
44
  - lr_scheduler: cosine