diff --git a/docs/reference/ptn_chained_model.md b/docs/reference/ptn_chained_model.md new file mode 100644 index 0000000..ca2ae03 --- /dev/null +++ b/docs/reference/ptn_chained_model.md @@ -0,0 +1,24 @@ +# Ptn Chained Model +Constructs a chained PyTorch model. + +### Fields: +- model_a: First model in the chain. +- model_b: Second model in the chain. +- closure: Optional differentiable function to be called at the end of forward. + +## Input +| Name | Data type | +|---|---| +| model_a | Ptmodel | +| model_b | Ptmodel | +| closure | Ptcallable | + +## Output +| Data type | +|---| +| Ptmodel | + +
+Category: PyTorch wrapper - Training + +ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved. diff --git a/docs/reference/pto_simple.md b/docs/reference/pto_simple.md new file mode 100644 index 0000000..264f9f1 --- /dev/null +++ b/docs/reference/pto_simple.md @@ -0,0 +1,20 @@ +# Instantiates the most basic optimizer to update W using the below formula +$$ +W -= lr * G +$$ + +## Input +| Name | Data type | +|---|---| +| model | Ptmodel | +| learning_rate | Float | + +## Output +| Data type | +|---| +| Ptoptimizer | + +
+Category: PyTorch wrapper - Training + +ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.