From 2f52351247666a8d09d755b636ec10f53696ad39 Mon Sep 17 00:00:00 2001 From: HowToSD Date: Mon, 10 Mar 2025 20:19:13 -0700 Subject: [PATCH] Updated model training document to add ResNet. --- docs/model_training.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/docs/model_training.md b/docs/model_training.md index c6eaaf4..4e68cde 100644 --- a/docs/model_training.md +++ b/docs/model_training.md @@ -1,8 +1,13 @@ # Model Training -Experimental PyTorch model training has been added. -No coding is required. Training is as simple as a single click! -Currently, only classification is supported. +You can traing a PyTorch model without any coding. +Training is as simple as a single click! +Currently, classification models are supported for following architecture (node names are in parentheses): +* Linear (Ptn Linear Model) +* Convolutional Network (Ptn Conv Model) +* ResNet (Ptn Resnet Model) + +You can adjust the model configuration including output dimension on the UI. The system is designed to be modular. For example, to switch from a linear model to a convolutional model, you only need to replace the model node (see screenshot below).