forbes-0023
363b49281b
- Project structure: solver/, freecad/, export/, configs/, scripts/, tests/, docs/ - pyproject.toml with dependency groups: core, train, freecad, dev - Hydra configs: dataset (synthetic, fusion360), model (baseline, gat), training (pretrain, finetune), export (production) - Dockerfile with CUDA+PyG GPU and CPU-only targets - docker-compose.yml for train, test, data-gen services - Makefile with targets: train, test, lint, format, type-check, data-gen, export, check - Pre-commit hooks: ruff, mypy, conventional commits - Gitea Actions CI: lint, type-check, test on push/PR - README with setup and usage instructions
29 lines
451 B
YAML
29 lines
451 B
YAML
# Advanced GAT model config
|
|
name: gat_solver
|
|
architecture: gat
|
|
|
|
encoder:
|
|
num_layers: 4
|
|
hidden_dim: 256
|
|
num_heads: 8
|
|
dropout: 0.1
|
|
residual: true
|
|
|
|
node_features_dim: 22
|
|
edge_features_dim: 22
|
|
|
|
heads:
|
|
edge_classification:
|
|
enabled: true
|
|
hidden_dim: 128
|
|
graph_classification:
|
|
enabled: true
|
|
num_classes: 4
|
|
joint_type:
|
|
enabled: true
|
|
num_classes: 12
|
|
dof_regression:
|
|
enabled: true
|
|
dof_tracking:
|
|
enabled: true
|