We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f072830 commit 4098372Copy full SHA for 4098372
advanced_source/neural_style_tutorial.py
@@ -291,8 +291,8 @@ def forward(self, input):
291
self.output = input
292
return self.output
293
294
- def backward(self, retain_variables=True):
295
- self.loss.backward(retain_variables=retain_variables)
+ def backward(self, retain_graph=True):
+ self.loss.backward(retain_graph=retain_graph)
296
return self.loss
297
298
@@ -363,8 +363,8 @@ def forward(self, input):
363
self.loss = self.criterion(self.G, self.target)
364
365
366
367
368
369
370
0 commit comments