bug fix

ni-chen · Jun 9, 2020 · a919f00 · a919f00
1 parent acc0fee
commit a919f00
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -65,7 +65,7 @@
  loss_criterion = nn.MSELoss()
 
  # optimizer = optim.SGD(model.parameters(), lr=0.0002, momentum=0.9)
- optimizer = optim.Adam(model.parameters(), lr=1e-4, betas=(0.5, 0.999))
+ optimizer = optim.Adam(params=filter(lambda p: p.requires_grad, model.parameters()), lr=1e-4, betas=(0.5, 0.999))
 
 
 

diff --git a/models/Resnet_model_1.py b/models/Resnet_model_1.py
@@ -168,7 +168,7 @@ def forward(self, imgs):
  output = self.residual_blocks(output) # (N, n_channels, w, h)
  output = self.conv_block2(output) # (N, n_channels, w, h)
  output = output + residual # (N, n_channels, w, h)
- output = self.subpixel_convolutional_blocks(output) # (N, n_channels, w * scaling factor, h * scaling factor)
+ #output = self.subpixel_convolutional_blocks(output) # (N, n_channels, w * scaling factor, h * scaling factor)
  out_imgs = self.conv_block3(output) # (N, 3, w * scaling factor, h * scaling factor)
 
  return out_imgs