Diffusers
TensorBoard
English
DDIMPipeline
dboshardy commited on
Commit
4a8dcc2
·
1 Parent(s): f0c2889

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ on the `huggan/smithsonian_butterflies_subset` dataset.
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 1e-05
40
- - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - gradient_accumulation_steps: 1
43
  - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None
 
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 1e-05
40
+ - train_batch_size: 32
41
  - eval_batch_size: 16
42
  - gradient_accumulation_steps: 1
43
  - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None