{"repo":"zaina-ml/ml_forge","free":true,"listed":false,"github":"https://github.com/zaina-ml/ml_forge","clone":"git clone https://github.com/zaina-ml/ml_forge.git","description":"A visual-based graph node editor for training computer vision models.","language":"Python","stars":421,"topics":["artificial-intelligence","beginner-friendly","computer-vision","data-science","deep-learning","desktop-app","drag-and-drop","gui","image-classification","machine-learning"],"license":"MIT","category":"machine-learning","readme_excerpt":"ML Forge A visual PyTorch pipeline editor. Build, train and run image classification models without writing code. --- MLForge v1.0.4 Additions: - Patched segfault error on Mac during template loading - Added logo to splash loading widnow - Added hints to node pallete when hovering over certain nodes. What it does - Visual pipeline - drag nodes onto a canvas, connect them with wires, and ML Forge generates and runs the training code for you - Three-tab workflow - Data Prep - Model - Training - Live training - watch loss curves update in real time, save checkpoints, run inference on your trained model - Export - export projects into clean PyTorch Tutorial Watch the MLForge tutorial video here: MLForge Tutorial --- Requirements IMPORTANT : PyTorch must be preinstalled for training, it is not installed as a dependency. - Python 3.10 or newer - PyTorch 2.0 or newer - torchvision GPU training is automatic if CUDA is available. CPU and Apple MPS are also supported. --- Quick Start To install MLForge, enter the following in your command prompt Then --- Cloning --- Building your first model 1. Data Prep tab - Add a Dataset node (MNIST, CIFAR10, CIFAR100, FashionMNIST, or ImageFolder) - Chain transforms : ToTensor is required, add Normalize for best results - End with a DataLoader (train) node - For proper validation, add a second chain (same dataset with train=False ) ending with DataLoader (val) 2. Model tab - Start with an Input node - shape is auto-filled from your dataset - Add la","default_branch":null,"files":null,"tree":[],"storefront":"/r/zaina-ml","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/zaina-ml/ml_forge/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}