From 14007d7eaa3e0047a6e93d248177c8c1d827b2ce Mon Sep 17 00:00:00 2001 From: David Nabergoj Date: Mon, 2 Sep 2024 20:59:03 +0200 Subject: [PATCH] Add architectures to __init__ --- torchflows/architectures.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/torchflows/architectures.py b/torchflows/architectures.py index a737ed0..13171f0 100644 --- a/torchflows/architectures.py +++ b/torchflows/architectures.py @@ -49,5 +49,8 @@ AffineGlow, RQSGlow, LRSGlow, - ShiftGlow + ShiftGlow, + DeepSigmoidGlow, + DeepDenseSigmoidGlow, + DenseSigmoidGlow )