From bf78d3b8b358c4be7a25f9f9438c842d837801fd Mon Sep 17 00:00:00 2001 From: kuangliu Date: Thu, 17 May 2018 15:19:48 +0800 Subject: [PATCH] Minor fix --- main.py | 1 - models/preact_resnet.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/main.py b/main.py index 7d49b829f..26a4e988d 100644 --- 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 index 13a88338b..abb1bc313 100644 --- 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()