| { | |
| "model": { | |
| "backbone": "Qwen/Qwen2.5-0.5B-Instruct", | |
| "decoder_dim": 128, | |
| "decoder_heads": 4, | |
| "decoder_layers": 2, | |
| "yaw_bins": 12, | |
| "max_objects": 128, | |
| "lora_rank": 8, | |
| "lora_alpha": 16, | |
| "backbone_dtype": "float32", | |
| "local_files_only": false | |
| }, | |
| "loss": { | |
| "position": 1.0, | |
| "size": 1.0, | |
| "yaw_cls": 1.0, | |
| "yaw_reg": 1.0, | |
| "hungarian": false, | |
| "alpha_position": 1.0, | |
| "alpha_size": 1.0, | |
| "box": 0.0, | |
| "collision": 0.0, | |
| "boundary": 0.0, | |
| "position_type": "l1", | |
| "size_type": "l1", | |
| "smooth_l1_beta": 1.0 | |
| }, | |
| "training": { | |
| "steps": 1000, | |
| "batch_size": 2, | |
| "learning_rate": 0.0001, | |
| "gradient_accumulation_steps": 1, | |
| "max_length": 8192, | |
| "seed": 42, | |
| "mixed_precision": "no", | |
| "checkpoint_every": 100, | |
| "validate_every": 100, | |
| "resume": null, | |
| "export_model_every_checkpoint": true | |
| }, | |
| "optimizer": { | |
| "warmup_steps": 30, | |
| "schedule": "cosine", | |
| "decoder_lr": 0.0001 | |
| }, | |
| "augmentation": { | |
| "rotate90": true, | |
| "mirror": false, | |
| "shuffle_objects": true, | |
| "drop_constraints_p": 0.3, | |
| "drop_support_p": 0.2, | |
| "category_only_description_p": 0.5 | |
| }, | |
| "validation": { | |
| "exclude_flags": [ | |
| "oob_objects", | |
| "fixed_collision", | |
| "overlapping_furniture" | |
| ] | |
| } | |
| } | |