Minor fix

kuangliu · May 17, 2018 · bf78d3b · bf78d3b
1 parent 010536e
commit bf78d3b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/main.py b/main.py
@@ -15,7 +15,6 @@
 
 from models import *
 from utils import progress_bar
-from torch.autograd import Variable
 
 
 parser = argparse.ArgumentParser(description='PyTorch CIFAR10 Training')

diff --git a/models/preact_resnet.py b/models/preact_resnet.py
@@ -112,7 +112,7 @@ def PreActResNet152():
 
 def test():
     net = PreActResNet18()
-    y = net((torch.randn(1,3,32,32))
+    y = net((torch.randn(1,3,32,32)))
     print(y.size())
 
 # test()