def __init__(self, scale, res_scale, conv=common.default_conv, n_feats=64):
    super(Net, self).__init__()
    n_resblock = 40
    kernel_size = 3
    act = nn.ReLU(True)
    m_head = [conv(n_feats, n_feats, kernel_size)]
    m_body = [common.ResBlock(conv, n_feats, kernel_size, act=act, res_scale=res_scale) for _ in range(n_resblock)]
    m_tail = [conv(n_feats, 3, kernel_size)]
    self.Net_head = nn.Sequential(*m_head)
    self.Net_body = nn.Sequential(*m_body)
    self.Net_tail = nn.Sequential(*m_tail)
def __init__(self, scale, res_scale, conv=common.default_conv, n_feats=64):
    super(Net, self).__init__()
    n_resblock = 40
    kernel_size = 3
    act = nn.ReLU(True)
    m_head = [conv(n_feats, n_feats, kernel_size)]
    m_body = [common.ResBlock(conv, n_feats, kernel_size, act=act, res_scale=res_scale) for _ in range(n_resblock)]
    m_tail = [conv(n_feats, 3, kernel_size)]
    self.Net_head = nn.Sequential(*m_head)
    self.Net_body = nn.Sequential(*m_body)
    self.Net_tail = nn.Sequential(*m_tail)