| { | |
| "backbone": "custom", | |
| "hf_repo": "ImageInpainters/liBERTy", | |
| "download_model": true, | |
| "upload_model": true, | |
| "upload_epochs": 5, | |
| "img_size": 128, | |
| "patch_size": 8, | |
| "batch_size": 16, | |
| "val_batch_size": 16, | |
| "num_workers": 2, | |
| "epochs": 20, | |
| "dim": 192, | |
| "depth": 6, | |
| "heads": 6, | |
| "mlp_ratio": 4.0, | |
| "dropout": 0.1, | |
| "decoder_dim": 192, | |
| "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": "google/vit-base-patch16-224-in21k", | |
| "freeze_pretrained_embeddings": true, | |
| "freeze_pretrained_blocks": 8, | |
| "train_dir": "/kaggle/input/coco-2017-dataset/coco2017/train2017", | |
| "val_dir": "/kaggle/input/coco-2017-dataset/coco2017/train2017", | |
| "output_dir": "outputs_inpaint_bert" | |
| } |