Update README.md
Browse files
README.md
CHANGED
@@ -3,7 +3,7 @@ This model was trained with the intent to classify whether or not an image conta
|
|
3 |
|
4 |
# Model Details
|
5 |
## How to Get Started with the Model
|
6 |
-
```
|
7 |
import torch
|
8 |
import transformers
|
9 |
|
@@ -30,8 +30,8 @@ with torch.no_grad():
|
|
30 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
31 |
```
|
32 |
# Training Details
|
33 |
-
Trained for three epochs
|
34 |
-
Learning rate: 5e-5
|
35 |
-
Optimizer: AdamW
|
36 |
-
Batch size: 64
|
37 |
-
Trained with FP32
|
|
|
3 |
|
4 |
# Model Details
|
5 |
## How to Get Started with the Model
|
6 |
+
```python
|
7 |
import torch
|
8 |
import transformers
|
9 |
|
|
|
30 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
31 |
```
|
32 |
# Training Details
|
33 |
+
- Trained for three epochs
|
34 |
+
- Learning rate: 5e-5
|
35 |
+
- Optimizer: AdamW
|
36 |
+
- Batch size: 64
|
37 |
+
- Trained with FP32
|