vatrpp / config.json
vittoriopippi
Error with model_type
e7e85b9
{
"_name_or_path": ".",
"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
}