Skip to content

Commit 4098372

Browse files
Naruto-Sasukechsasank
authored andcommitted
Update neural_style_tutorial.py (pytorch#166)
replace `retain_variables` with `retain_graph`
1 parent f072830 commit 4098372

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

advanced_source/neural_style_tutorial.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -291,8 +291,8 @@ def forward(self, input):
291291
self.output = input
292292
return self.output
293293

294-
def backward(self, retain_variables=True):
295-
self.loss.backward(retain_variables=retain_variables)
294+
def backward(self, retain_graph=True):
295+
self.loss.backward(retain_graph=retain_graph)
296296
return self.loss
297297

298298

@@ -363,8 +363,8 @@ def forward(self, input):
363363
self.loss = self.criterion(self.G, self.target)
364364
return self.output
365365

366-
def backward(self, retain_variables=True):
367-
self.loss.backward(retain_variables=retain_variables)
366+
def backward(self, retain_graph=True):
367+
self.loss.backward(retain_graph=retain_graph)
368368
return self.loss
369369

370370

0 commit comments

Comments
 (0)