shalpin87
/

diffusion_cifar

Model card Files Files and versions

Sean Halpin commited on Aug 18, 2022

Commit

2646413

·

1 Parent(s): bbb4872

update model card README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ on the `CIFAR10` dataset.
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
 - train_batch_size: 256
-- eval_batch_size: 256
 - 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: 1e-05
 - train_batch_size: 256
+- 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