test_pickable_sequential_chain.py 文件源码

python
阅读 19 收藏 0 点赞 0 评论 0

项目:chainercv 作者: chainer 项目源码 文件源码
def check_call(self, x, expects):
        outs = self.link(x)

        if isinstance(self.pick, tuple):
            pick = self.pick
        else:
            if self.pick is None:
                pick = ('l2',)
            else:
                pick = (self.pick,)
            outs = (outs,)

        self.assertEqual(len(outs), len(pick))

        for out, layer_name in zip(outs, pick):
            self.assertIsInstance(out, chainer.Variable)
            self.assertIsInstance(out.array, self.link.xp.ndarray)

            out = to_cpu(out.array)
            np.testing.assert_equal(out, to_cpu(expects[layer_name].array))
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号