Update model.py

model.py

@@ -310,6 +310,8 @@ class ParallelGatedConvBlock(nn.Module):
 
         z, inference_params = self.filter(z, inference_params=inference_params, padding_mask=padding_mask)
 
+        self.filter_output = z
+
         z_in = self.out_filter_dense(z) + u
 
         if type(padding_mask) == torch.Tensor:  # guard against bias