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
26 lines
434 B
YAML
26 lines
434 B
YAML
# Production model export config
|
|
model_checkpoint: checkpoints/finetune/best_val_loss.ckpt
|
|
output_dir: export/
|
|
|
|
formats:
|
|
onnx:
|
|
enabled: true
|
|
opset_version: 17
|
|
dynamic_axes: true
|
|
torchscript:
|
|
enabled: true
|
|
|
|
model_card:
|
|
version: "0.1.0"
|
|
architecture: baseline
|
|
training_data:
|
|
- synthetic_100k
|
|
- fusion360_gallery
|
|
|
|
size_budget_mb: 50
|
|
|
|
inference:
|
|
device: cpu
|
|
batch_size: 1
|
|
confidence_threshold: 0.8
|