Train and Run Machine Learning Models In The Browser

Train AI Models In Your Browser

Load your image dataset to start training your image classification model.

Want to resume from a project you saved to your device?

Load project.zip

Select Training Data

Choose the image categories for the vision model to train on.

Load image dataset

1 / 1

Model Layers

Build the model by stacking together a list of layers to form a neural network.

Click here to add layers

Set Hyperparameters

Hyperparameters determine the type of model and how effectively it is trained. They are frequently adjusted to help the model reach its optimal state.

Training epochs

Batch size

Optimizer

Learning rate

Loss function

Validation split

Metrics (optional)

Train Model

Train the model until the accuracy metric is as close to 100% and loss metric to zero as possible. A WebGPU supported browser is recommended for faster training on complex models. To prevent interruptions, keep this tab in the foreground. Browsers may throttle or sleep tabs not in view.

Epoch 0·Batch 0/0·00:00:00

Status: Model is idle

Inference

Classify images directly in the browser

Load images

Drag and drop or paste images here for classification