miguelcarv commited on
Commit
ebce8b8
·
verified ·
1 Parent(s): 00e547e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
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