update model card README.md

README.md

@@ -73,7 +73,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0005
+- learning_rate: 0.1
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
@@ -87,7 +87,7 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.20.1
+- Transformers 4.24.0
 - Pytorch 1.11.0+cu113
 - Datasets 2.5.1
 - Tokenizers 0.11.6
@@ -161,10 +161,10 @@ The following hyperparameters were used during training:
               'fp16': True,
               'hub_model_id': 'tomekkorbak/detoxify_toxicity',
               'hub_strategy': 'all_checkpoints',
-              'learning_rate': 0.0005,
+              'learning_rate': 0.1,
               'logging_first_step': True,
               'logging_steps': 1,
-              'num_tokens': 3300000000,
+              'num_tokens': 3300000000.0,
               'output_dir': 'training_output104340',
               'per_device_train_batch_size': 8,
               'push_to_hub': True,
@@ -176,4 +176,4 @@ The following hyperparameters were used during training:
               'weight_decay': 0.1}}
 
 # Wandb URL:
-https://wandb.ai/tomekkorbak/apo/runs/1kyuw9ou
+https://wandb.ai/tomekkorbak/apo/runs/1l6hdjln