PCBWorld checkpoints

Trained routing policies from the PCBWorld paper. Weights only. All are decoder-only transformer policies (4 layers, d_model 128, 8 heads, FFN 512; 1.6M parameters) trained from scratch, seeds 42–45.

Folder Dataset Agent
d1/grid{10,50,100,200,500}/ D1-10 … D1-500 (synthetic grid-based boards) PPO
d2/ppo/ D2 (synthetic gridless boards) PPO
d2/ppo_terminal/ D2 PPO (terminal)
d2/grpo/ D2 GRPO
d2/ppo_no_finish/ D2 PPO (w/o finish)
d2/reward_ablation/wire<λw>_via<λv>/ D2 PPO, wirelength weight λw × via weight λv

d2/reward_ablation/wire0.002_via0.1/ is the default weighting and holds the same files as d2/ppo/. Each file holds the weights (policy_state_dict) and the training arguments (args).

hf download LG-AI-Research/PCBWorld --local-dir var/checkpoints/pcbworld

License

EXAONE AI Model License Agreement 1.2 - NC (LICENSE): research and educational use only. Commercial use of the model and of its output, including boards routed with it, is not permitted. The PCBWorld code is licensed separately; see its repository.

Downloads last month

-

Downloads are not tracked for this model. How to track
Video Preview
loading

Paper for LG-AI-Research/PCBWorld