BluebrainAI
/

duo-predict-gpt2-medium-wikitext

Feature Extraction

duo-predict-gpt2

Generated from Trainer

Model card Files Files and versions

shivanandmn commited on Apr 1

Commit

5404364

·

verified ·

1 Parent(s): 2cab8fe

Model save

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,8 +32,8 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
-- train_batch_size: 8
-- eval_batch_size: 8
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear

 The following hyperparameters were used during training:
 - learning_rate: 0.0001
+- train_batch_size: 64
+- eval_batch_size: 64
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear