From 4eed286a5d8ad79a08fd1623e0752364f6b6801c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=97=E5=A4=A7=E6=9D=83?= Date: Thu, 10 May 2018 22:59:03 +0800 Subject: [PATCH] error with filter Reference: https://stackoverflow.com/questions/15876259/typeerror-filter-object-is-not-subscriptable?utm_medium=organic&utm_source=google_rich_qa&utm_campaign=google_rich_qa --- models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index d7839bc..03eff70 100644 --- a/models.py +++ b/models.py @@ -256,7 +256,7 @@ def _construct_diff_model(self, base_model, keep_rgb=False): # Torch models are usually defined in a hierarchical way. # nn.modules.children() return all sub modules in a DFS manner modules = list(self.base_model.modules()) - first_conv_idx = filter(lambda x: isinstance(modules[x], nn.Conv2d), list(range(len(modules))))[0] + first_conv_idx = list(filter(lambda x: isinstance(modules[x], nn.Conv2d), list(range(len(modules)))))[0] conv_layer = modules[first_conv_idx] container = modules[first_conv_idx - 1] @@ -301,4 +301,4 @@ def get_augmentation(self): GroupRandomHorizontalFlip(is_flow=True)]) elif self.modality == 'RGBDiff': return torchvision.transforms.Compose([GroupMultiScaleCrop(self.input_size, [1, .875, .75]), - GroupRandomHorizontalFlip(is_flow=False)]) \ No newline at end of file + GroupRandomHorizontalFlip(is_flow=False)])