summaryrefslogtreecommitdiff
path: root/models/test_model.py
diff options
context:
space:
mode:
authorJun-Yan Zhu <junyanz@users.noreply.github.com>2017-11-09 13:19:23 -0800
committerGitHub <noreply@github.com>2017-11-09 13:19:23 -0800
commit7aba5d9440b123bb1a151cbd62cf61ff0d733c41 (patch)
tree9621879f1070cf1d99829fa020e87000f878a3fa /models/test_model.py
parenta24e24d67d88f75869f447690f7d994fe7d42e2d (diff)
parentc2fc8d442f1248231eab4b73e111665288b1e615 (diff)
Merge pull request #146 from SsnL/update_pytorch_syntax
PyTorch best practices
Diffstat (limited to 'models/test_model.py')
-rw-r--r--models/test_model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/test_model.py b/models/test_model.py
index 4af1fe1..2ae2812 100644
--- a/models/test_model.py
+++ b/models/test_model.py
@@ -34,7 +34,7 @@ class TestModel(BaseModel):
def test(self):
self.real_A = Variable(self.input_A)
- self.fake_B = self.netG.forward(self.real_A)
+ self.fake_B = self.netG(self.real_A)
# get image paths
def get_image_paths(self):