vatr/config.json

{
  "add_noise": false,
  "alphabet": "Only thewigsofrcvdampbkuq.A-210xT5'MDL,RYHJ\"ISPWENj&BC93VGFKz();#:!7U64Q8?+*ZX/%",
  "architectures": [
    "VATrPP"
  ],
  "auto_map": {
    "AutoConfig": "blowing-up-groundhogs/vatrpp--configuration_vatrpp.VATrPPConfig",
    "AutoModel": "blowing-up-groundhogs/vatrpp--modeling_vatrpp.VATrPP"
  },
  "augment_ocr": false,
  "batch_size": 8,
  "corpus": "standard",
  "d_crop_size": null,
  "d_lr": 1e-05,
  "dataset": "IAM",
  "device": "cuda",
  "english_words_path": "files/english_words.txt",
  "epochs": 100000,
  "feat_model_path": "files/resnet_18_pretrained.pth",
  "file_suffix": null,
  "g_lr": 5e-05,
  "img_height": 32,
  "is_cycle": false,
  "label_encoder": "default",
  "model_type": "vatrpp",
  "no_ocr_loss": false,
  "no_writer_loss": false,
  "num_examples": 15,
  "num_words": 3,
  "num_workers": 0,
  "num_writers": 339,
  "ocr_lr": 5e-05,
  "query_input": "unifont",
  "resolution": 16,
  "save_model": 5,
  "save_model_history": 500,
  "save_model_path": "saved_models",
  "seed": 742,
  "special_alphabet": "\u0391\u03b1\u0392\u03b2\u0393\u03b3\u0394\u03b4\u0395\u03b5\u0396\u03b6\u0397\u03b7\u0398\u03b8\u0399\u03b9\u039a\u03ba\u039b\u03bb\u039c\u03bc\u039d\u03bd\u039e\u03be\u039f\u03bf\u03a0\u03c0\u03a1\u03c1\u03a3\u03c3\u03c2\u03a4\u03c4\u03a5\u03c5\u03a6\u03c6\u03a7\u03c7\u03a8\u03c8\u03a9\u03c9",
  "tag": "debug",
  "text_aug_type": "proportional",
  "text_augment_strength": 0.0,
  "torch_dtype": "float32",
  "transformers_version": "4.46.2",
  "vocab_size": 80,
  "w_lr": 5e-05,
  "wandb": false,
  "writer_loss_weight": 1.0
}