File size: 1,327 Bytes
e6f80f3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
{
  "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"
    ]
  }
}