nawage
/

dragons-test

Model card Files Files and versions Metrics Training metrics Community

nawage commited on Aug 31, 2022

Commit

b6e07b6

·

1 Parent(s): 366084d

update model card README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -37,8 +37,8 @@ on the `imagefolder` dataset.
 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
 - lr_scheduler: None

 The following hyperparameters were used during training:
 - learning_rate: 0.0001
+- train_batch_size: 4
+- eval_batch_size: 4
 - gradient_accumulation_steps: 1
 - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None
 - lr_scheduler: None