File size: 382 Bytes
d6b08e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# OCR Model
This is an OCR model trained with PyTorch Lightning.
## Files
- `final_model.pt`: Original model file
- `pytorch_model.bin`: Same model with HuggingFace naming convention
## Usage
```python
import torch
# Load using original filename
model = torch.load("final_model.pt")
# Or using HuggingFace convention
model = torch.load("pytorch_model.bin")
model.eval()
```
|