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