DesigNet / config.json
iago-suarez's picture
Push model using huggingface_hub.
4e348c2 verified
{
"_class_path": "fontia.genai.model.lightning_module.FontConditionalSVGVAE",
"_instantiator": "lightning.pytorch.cli.instantiate_module",
"aggregator_transformer": true,
"args_dim": 256,
"checkpoint_path": null,
"cross_reconstruction_multiplier": 1.5,
"d_model": 512,
"decoding_glyphs": [
"A",
"B",
"C",
"c",
"D",
"d",
"E",
"F",
"f",
"G",
"h",
"I",
"i",
"J",
"j",
"K",
"k",
"L",
"l",
"M",
"N",
"n",
"O",
"o",
"P",
"p",
"Q",
"q",
"R",
"S",
"s",
"T",
"t",
"U",
"V",
"v",
"W",
"w",
"X",
"x",
"Y",
"y",
"Z",
"z"
],
"dim_feedforward": 1024,
"dim_z": 512,
"discrete_arguments": false,
"dropout": 0.1,
"encoding_glyphs": [
"H",
"a",
"m",
"b",
"u",
"r",
"g",
"e"
],
"eval_frequency": 10,
"hierarchical": true,
"learning_rate": 0.0001,
"loss_alignment_weight": 1.0,
"loss_args_weight": 10000,
"loss_aux_weight": 5000,
"loss_cmd_weight": 1.0,
"loss_consistency_weight": 5000,
"loss_continuity_weight": 1.0,
"loss_visibility_weight": 1.0,
"max_evaluation_batches": 10,
"max_num_groups": 4,
"max_seq_len": 32,
"max_total_len": 64,
"n_args": 8,
"n_commands": 5,
"n_heads": 16,
"n_layers": 10,
"n_layers_decode": 10,
"normalized_coords": true,
"num_decode_cross_set": null,
"num_decode_ref_set": null,
"num_decoding_glyphs": 44,
"num_encoding_glyphs": 8,
"num_groups_proposal": 4,
"predict_alignment": true,
"predict_continuity": true,
"rel_targets": false,
"train_with_refinement": false,
"use_aux_points": true,
"use_point_posenc": false,
"warmup_steps": 500,
"weight_decay": 0.0,
"weights": null
}