From 07ddc54192d2ef6d12dc6967003f9db795fb5e13 Mon Sep 17 00:00:00 2001 From: JohnMark Taylor Date: Thu, 17 Oct 2024 14:50:27 -0400 Subject: [PATCH] Updated readme and setup.py after pull request. --- tests/test_validation_and_visuals.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_validation_and_visuals.py b/tests/test_validation_and_visuals.py index 9ac2ebf..65881ec 100644 --- a/tests/test_validation_and_visuals.py +++ b/tests/test_validation_and_visuals.py @@ -582,6 +582,7 @@ def test_buffer_rewrite_model(): show_model_graph( model, model_input, + save_only=True, vis_opt="unrolled", vis_outpath=opj("visualization_outputs", "toy-networks", "buffer_rewrite_model_visible_nested_unrolled"), vis_buffer_layers=True, @@ -3107,9 +3108,9 @@ class OneHotAutoEncoder(L.LightningModule): def __init__(self): super().__init__() self.model = nn.Sequential( - nn.Linear(16, 4), - nn.ReLU(), - nn.Linear(4, 16), + nn.Linear(16, 4), + nn.ReLU(), + nn.Linear(4, 16), ) def forward(self, x): @@ -3127,4 +3128,3 @@ def forward(self, x): vis_outpath=opj("visualization_outputs", "lightning", "one-hot-autoencoder"), ) assert validate_saved_activations(model, model_inputs) -