def __init__(self, Encoder, num_features, block_channel): super(model, self).__init__() self.E = Encoder self.D = modules.D(num_features) self.MFF = modules.MFF(block_channel) self.R = modules.R(block_channel)
def __init__(self, Encoder, num_features, block_channel, refinenet): super(model, self).__init__() self.E = Encoder self.D = modules.D(num_features) self.MFF = modules.MFF(block_channel) self.R = R_CLSTM_modules(block_channel)