def _num_filters_stub(block_id): return networks.num_filters(block_id, 8, 1, 8)
def _num_filters_fn(block_id): """Computes number of filters of block `block_id`.""" return networks.num_filters(block_id, kwargs['fmap_base'], kwargs['fmap_decay'], kwargs['fmap_max'])
def test_num_filters(self): self.assertEqual(networks.num_filters(1, 4096, 1, 256), 256) self.assertEqual(networks.num_filters(5, 4096, 1, 256), 128)