config.json

{
  "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"
}