OrderedDict not needed

This commit is contained in:
James Betker 2020-07-04 14:09:27 -06:00
parent 9048105b72
commit 0ee39d419b

View File

@ -3,7 +3,6 @@ from torch import nn
from switched_conv import BareConvSwitch, compute_attention_specificity from switched_conv import BareConvSwitch, compute_attention_specificity
import torch.nn.functional as F import torch.nn.functional as F
import functools import functools
from collections import OrderedDict
from models.archs.arch_util import initialize_weights from models.archs.arch_util import initialize_weights
from switched_conv_util import save_attention_to_image from switched_conv_util import save_attention_to_image