From 95e9f58c4f5c80c550f7270ad5c4b17a6b15dac9 Mon Sep 17 00:00:00 2001 From: HowToSD Date: Fri, 14 Mar 2025 19:43:13 -0700 Subject: [PATCH] Fixed doc generator to generate doc for a few nodes which did not have doc --- dev/scripts/reference_generate.py | 2 +- docs/reference/node_reference.md | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/dev/scripts/reference_generate.py b/dev/scripts/reference_generate.py index 11f5f3f..a86053a 100644 --- a/dev/scripts/reference_generate.py +++ b/dev/scripts/reference_generate.py @@ -108,7 +108,7 @@ def process_python_file(file_path: str, output_dir: str, node_list: Dict[str, Li if isinstance(node, ast.ClassDef): class_info = extract_class_info(node) if not class_info: - return + continue input_section = extract_input_types(node) output_section = extract_return_types(node) diff --git a/docs/reference/node_reference.md b/docs/reference/node_reference.md index 3ddd10c..bdf645c 100644 --- a/docs/reference/node_reference.md +++ b/docs/reference/node_reference.md @@ -155,6 +155,7 @@ ## PyTorch wrapper - Training | Node | Description | | --- | --- | +| [Instantiates the most basic optimizer to update W using the below formula](pto_simple.md) | $$ | | [Pt Apply Function](pt_apply_function.md) | Applies a function to the input tensor. | | [Pt Data Loader](pt_data_loader.md) | Loads data from a dataset node and creates a PyTorch DataLoader. | | [Pt Data Loader From Tensors](pt_data_loader_from_tensors.md) | Creates a Torchvision Dataloader from a pair of tensors. | @@ -165,6 +166,7 @@ | [Pt Train Classification Model](pt_train_classification_model.md) | Trains a classification model using a given dataset, optimizer, and number of epochs. | | [Pt Train Classification Model Lr](pt_train_classification_model_lr.md) | Trains a classification model using a given dataset, optimizer, and number of epochs with learning rate decay. | | [Pt Train Regression Model](pt_train_regression_model.md) | Trains a regression model using a given dataset, optimizer, and number of epochs. | +| [Ptn Chained Model](ptn_chained_model.md) | Constructs a chained PyTorch model. | | [Ptn Conv Model](ptn_conv_model.md) | A convolutional model consisting of multiple convolutional layers. | | [Ptn Linear](ptn_linear.md) | A linear model consisting of a single dense layer. | | [Ptn Linear Model](ptn_linear_model.md) | A linear model consisting of dense layers. |