diff --git a/training/training_scripts/NN_Filtering_HOP/model/model.py b/training/training_scripts/NN_Filtering_HOP/model/model.py index edf277260dbcbe3e1f5cf7bdbd6df50cfce1e681..f58d00fbfea135340d7d914f10a31bc535a35946 100644 --- a/training/training_scripts/NN_Filtering_HOP/model/model.py +++ b/training/training_scripts/NN_Filtering_HOP/model/model.py @@ -208,7 +208,7 @@ class SADLNet(nn.Sequential): Conv(D6, C, kernel_size=3, stride=2), *[ResidualBlock(C, C1, C21, C22, C31) for _ in range(N)], Conv(C, C, kernel_size=3), - Conv(C, output_channels, kernel_size=3), + Conv(C, output_channels, kernel_size=3, post_activation=None), ) def get_example_inputs(