deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix torchvision import bug

Browse Source
This commit is contained in:
James Betker 2020-09-24 17:38:34 -06:00
parent 58886109d4
commit 553917a8d1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/archs/SPSR_arch.py
View File

@ -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