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 = refinenet_dict[refinenet](block_channel)