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 |
|