Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ metrics:
|
|
17 |
|
18 |
## Model description
|
19 |
|
20 |
-
`bert-base-uncased` finetuned on the AG News dataset using PyTorch Lightning. Sequence length 128, learning rate 2e-5, batch size 32, 4 T4 GPUs, 4 epochs. [The code can be found here](https://github.com/nateraw)
|
21 |
|
22 |
#### Limitations and bias
|
23 |
|
17 |
|
18 |
## Model description
|
19 |
|
20 |
+
`bert-base-uncased` finetuned on the AG News dataset using PyTorch Lightning. Sequence length 128, learning rate 2e-5, batch size 32, 4 T4 GPUs, 4 epochs. [The code can be found here](https://github.com/nateraw/hf-text-classification)
|
21 |
|
22 |
#### Limitations and bias
|
23 |
|