Update generator_2.py

generator_2.py

@@ -96,7 +96,7 @@ class discrim_block(nn.Module):
         return out
 
 
-class Generator_2(nn.Module):
+class Generator(nn.Module):
 
     def __init__(self, img_feat = 3, n_feats = 64, kernel_size = 3, num_block = 16, act = nn.PReLU(), scale=4):
         super(Generator, self).__init__()