BroLaurens commited on
Commit
5c77061
1 Parent(s): 0296e8f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -62,6 +62,7 @@ predicted_labels = [[int2str[x] for x in t] for t in predictions.tolist()]
62
 
63
  The model was trained using the following hyperparameters:
64
 
 
65
  base_model: distilbert/distilbert-base-uncased
66
  learning_rate: 2e-5
67
  batch_size: 32
@@ -70,4 +71,5 @@ optimizer: adamw
70
  adam_beta1: 0.9
71
  adam_beta2: 0.999
72
  adam_epsilon: 1e-08
73
- loss_function: cross entropy loss
 
 
62
 
63
  The model was trained using the following hyperparameters:
64
 
65
+ ```
66
  base_model: distilbert/distilbert-base-uncased
67
  learning_rate: 2e-5
68
  batch_size: 32
 
71
  adam_beta1: 0.9
72
  adam_beta2: 0.999
73
  adam_epsilon: 1e-08
74
+ loss_function: cross entropy loss
75
+ ```