| { | |
| "backbone": "hf_vit", | |
| "hf_repo": "ImageInpainters/ViTInpainter", | |
| "download_model": true, | |
| "upload_model": true, | |
| "upload_epochs": 1, | |
| "img_size": 224, | |
| "patch_size": 16, | |
| "batch_size": 4, | |
| "val_batch_size": 4, | |
| "num_workers": 2, | |
| "epochs": 20, | |
| "dim": 192, | |
| "depth": 6, | |
| "heads": 6, | |
| "mlp_ratio": 4.0, | |
| "dropout": 0.1, | |
| "decoder_dim": 256, | |
| "lr": 0.0002, | |
| "encoder_lr_scale": 0.1, | |
| "weight_decay": 0.0001, | |
| "valid_weight": 0.05, | |
| "boundary_weight": 0.25, | |
| "boundary_kernel_size": 7, | |
| "pretrained_model_name": "/kaggle/input/datasets/rjawesome/pretrained-vit/vit_base_patch16_224", | |
| "freeze_pretrained_embeddings": true, | |
| "freeze_pretrained_blocks": 10, | |
| "train_dir": "/kaggle/input/datasets/awsaf49/coco-2017-dataset/coco2017/train2017", | |
| "val_dir": "/kaggle/input/datasets/awsaf49/coco-2017-dataset/coco2017/val2017", | |
| "output_dir": "/kaggle/working/outputs_inpaint_bert" | |
| } |