Update README.md
Browse files
README.md
CHANGED
|
@@ -20,6 +20,8 @@ This model is a VisionEncoderDecoderModel fine-tuned on a dataset for generating
|
|
| 20 |
- **Decoder**: GPT-2
|
| 21 |
- **Framework**: PyTorch
|
| 22 |
- **DDP (Distributed Data Parallel)**: Used for training
|
|
|
|
|
|
|
| 23 |
|
| 24 |
## Training Data
|
| 25 |
|
|
|
|
| 20 |
- **Decoder**: GPT-2
|
| 21 |
- **Framework**: PyTorch
|
| 22 |
- **DDP (Distributed Data Parallel)**: Used for training
|
| 23 |
+
|
| 24 |
+
<img src="https://github.com/d-gurgurov/im2latex/blob/main/assets/im2latex.png?raw=true" alt="architecture" width="700"/>
|
| 25 |
|
| 26 |
## Training Data
|
| 27 |
|