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