File size: 556 Bytes
7f56ec4 1fba0e8 7f56ec4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | ---
license: mit
library_name: pytorch
tags:
- image-inpainting
- masked-image-modeling
- bidirectional-transformer
- custom-model
---
# Better Inpainting Transformer
This repository contains the improved inpainting model.
## Files
- `pytorch_model.bin`: inference weights
- `config.json`: config
- `training_metadata.json`: epoch and val loss for best uploaded checkpoint
- `model_last.pt`: resumable training checkpoint
- `model_best.pt`: best local checkpoint
## Notes
Recreate the `BetterInpaintTransformer` class and load the state dict manually.
|