DesigNet / vae /config.json
iago-suarez's picture
Upload folder using huggingface_hub
5bb91e9 verified
{
"_class_path": "fontia.deepsvg.deepsvg.model.lightning_module.SVG_VAE",
"_instantiator": "lightning.pytorch.cli.instantiate_module",
"args_dim": 256,
"close_paths": true,
"d_model": 256,
"dim_feedforward": 512,
"dim_z": 256,
"discrete_arguments": false,
"dropout": 0.1,
"eval_frequency": 10,
"hierarchical": true,
"learning_rate": 0.0001,
"loss_alignment_weight": 1,
"loss_args_weight": 12500,
"loss_aux_weight": 5000,
"loss_cmd_weight": 1,
"loss_consistency_weight": 5000,
"loss_continuity_weight": 1,
"loss_visibility_weight": 1,
"max_evaluation_batches": 10,
"max_num_groups": 4,
"max_seq_len": 32,
"max_total_len": 64,
"n_args": 8,
"n_commands": 5,
"n_heads": 8,
"n_layers": 4,
"n_layers_decode": 4,
"normalized_coords": true,
"num_glyphs": 62,
"num_groups_proposal": 4,
"predict_alignment": true,
"predict_continuity": true,
"rel_targets": false,
"train_with_refinement": false,
"use_aux_points": true,
"use_letter_encoding": false,
"use_point_posenc": false,
"weight_decay": 0.0
}