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
25 lines
411 B
YAML
25 lines
411 B
YAML
# Baseline GIN model config
|
|
name: baseline
|
|
architecture: gin
|
|
|
|
encoder:
|
|
num_layers: 3
|
|
hidden_dim: 128
|
|
dropout: 0.1
|
|
|
|
node_features_dim: 22
|
|
edge_features_dim: 22
|
|
|
|
heads:
|
|
edge_classification:
|
|
enabled: true
|
|
hidden_dim: 64
|
|
graph_classification:
|
|
enabled: true
|
|
num_classes: 4 # rigid, under, over, mixed
|
|
joint_type:
|
|
enabled: true
|
|
num_classes: 12
|
|
dof_regression:
|
|
enabled: true
|