From 0ee39d419bd72bb1c98d58f8f55ac77b0225624c Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 4 Jul 2020 14:09:27 -0600 Subject: [PATCH] OrderedDict not needed --- codes/models/archs/SRG1_arch.py | 1 - 1 file changed, 1 deletion(-) diff --git a/codes/models/archs/SRG1_arch.py b/codes/models/archs/SRG1_arch.py index 089d0594..e30a0181 100644 --- a/codes/models/archs/SRG1_arch.py +++ b/codes/models/archs/SRG1_arch.py @@ -3,7 +3,6 @@ from torch import nn from switched_conv import BareConvSwitch, compute_attention_specificity import torch.nn.functional as F import functools -from collections import OrderedDict from models.archs.arch_util import initialize_weights from switched_conv_util import save_attention_to_image