Solved by Bled_Clement in post #3 
              
              
                Hi Peter, 
I’ve since made the following changes to my network: 
import torch
import torch.nn as nn
from wiener_3d import wiener_3d


class WienerFilter(nn.Module):
    def __init__(self, param_a=0.05, param_b=10):
        super(WienerFilter, self).__init__()
        self.param_a = nn.Parameter(torc…