From 553917a8d1d11b75ff2f27f1c79cdb2cde953bf6 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Thu, 24 Sep 2020 17:38:34 -0600
Subject: [PATCH] Fix torchvision import bug

---
 codes/models/archs/SPSR_arch.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/codes/models/archs/SPSR_arch.py b/codes/models/archs/SPSR_arch.py
index 047aec22..8aeb4e01 100644
--- a/codes/models/archs/SPSR_arch.py
+++ b/codes/models/archs/SPSR_arch.py
@@ -10,6 +10,7 @@ from switched_conv_util import save_attention_to_image_rgb
 from switched_conv import compute_attention_specificity
 import functools
 import os
+import torchvision
 
 
 class ImageGradient(nn.Module):
@@ -674,3 +675,5 @@ class Spsr5(nn.Module):
             val["switch_%i_specificity" % (i,)] = means[i]
             val["switch_%i_histogram" % (i,)] = hists[i]
         return val
+
+