diff --git a/chainercv/links/model/pickable_sequential_chain.py b/chainercv/links/model/pickable_sequential_chain.py index 178d839cf9..bca032698a 100644 --- a/chainercv/links/model/pickable_sequential_chain.py +++ b/chainercv/links/model/pickable_sequential_chain.py @@ -143,6 +143,7 @@ def __call__(self, x): h = x for name in self.layer_names[:last_index + 1]: h = self[name](h) + h.name = name if name in pick: layers[name] = h diff --git a/tests/links_tests/model_tests/test_pickable_sequential_chain.py b/tests/links_tests/model_tests/test_pickable_sequential_chain.py index 10842b3694..1b4e2b3b4e 100644 --- a/tests/links_tests/model_tests/test_pickable_sequential_chain.py +++ b/tests/links_tests/model_tests/test_pickable_sequential_chain.py @@ -60,6 +60,7 @@ def check_call(self, x, expects): for out, layer_name in zip(outs, pick): self.assertIsInstance(out, chainer.Variable) self.assertIsInstance(out.array, self.link.xp.ndarray) + self.assertEqual(out.name, layer_name) out = to_cpu(out.array) np.testing.assert_equal(out, to_cpu(expects[layer_name].array))