structcast-model

What problem does it solve?

This Skill automates the process of generating PyTorch, Flax, and Keras models from YAML templates, enabling users to create and train models declaratively without writing extensive code.

Core Features & Use Cases

• Model Generation: Generate PyTorch, Flax, and Keras models from YAML templates.
• Training Workflow: Generate PyTorch training workflows and CLI commands.
• Configuration: Describe model architecture, optimizer logic, dataset configuration, and training orchestration declaratively in YAML templates.
• Use Case: If you need to create a new ConvNeXtV2 model for PyTorch, you can use this Skill to generate the necessary model code from a YAML template.

Quick Start

Run the following command to generate a PyTorch model from a YAML template:

scm torch create model cfg/torch/models/ConvNeXtV2.yaml