PyTorch Module

PyTorch Generator

Generate a ready-to-run PyTorch project in seconds.

Minimal PyTorch project with training and prediction flow.

Generation Options

Review the config values for your PyTorch project.

Name

Input Dim Output Dim

Hidden Activation Metrics

Learning Rate Epochs

Batch Size Use MLflow logging (`prod: true`)

Workflow Output

Copy the setup snippet and run the commands in your CLI.

Select options and generate a command.

Install

pip install genxcode

Config

# Generated by `genxcode init`
name: pytorch
arch: mlp
input_dim: 784
output_dim: 10
hidden:
  - 256
  - 128
activation: relu
metrics:
  - accuracy
  - loss
prod: false
lr: 0.001
epochs: 10
batch_size: 32

Workflow

genxcode init pytorch
# edit genxcode.yaml
genxcode apply

Generated Files

pytorch/
    src/
        __init__.py
        config.py
        models.py
        data.py
        train.py
        predict.py
    main.py
    requirements.txt