← all repositories

zaina-ml/ml_forge

A drag-and-drop node editor that generates and runs PyTorch training code for image classification models.

ml_forge
Velocity · 7d
+4.7
★ / day
Trend
steady
star history

ML Forge is a visual desktop application that provides a three-tab workflow (Data Prep, Model, Training) for building computer vision pipelines. Users drag nodes onto a canvas, connect them with wires, and the tool automatically generates PyTorch training code. It supports popular datasets (MNIST, CIFAR10, CIFAR100, FashionMNIST, ImageFolder) and provides live training visualization with loss curves, checkpoint saving, and model inference. The tool exports projects into clean, runnable PyTorch code.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.