sklearn dependency was not explicitly mentioned in the README. To make the code working on recent pytorch, I needed to update in line of code :
@@ -92,7 +92,7 @@ class vgg19(nn.Module):
deconvolved_feature_forward = self.forward(level=src_level, start_level=dst_level, set_as_var = False)
loss_perceptual = criterionPerceptual(deconvolved_feature_forward, src_layer)
loss_perceptual.backward()
- error = loss_perceptual.data[0]
+ error = loss_perceptual.item()
self.update_last_losses(error)
if (i % 3 == 0) and (print_errors == True):
print("error: ", error)If you want to contribute with another review, please follow these instructions.
Please consider to cut/paste/edit the raw JSON data attached to this paper.