uumlaut
/

ddpm-vangogh

Model card Files Files and versions

Metrics Training metrics Community

uumlaut commited on Jan 6, 2023

Commit

4a4dd40

·

1 Parent(s): 7bd233f

update model card README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,8 +36,8 @@ on the `imagefolder` dataset.
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
-- train_batch_size: 16
 - eval_batch_size: 16
 - gradient_accumulation_steps: 1
 - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None

 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 1e-06
+- train_batch_size: 2
 - eval_batch_size: 16
 - gradient_accumulation_steps: 1
 - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None