def setUp(self):
if self.n_layers == 50:
self.link = resnet.ResNet50Layers(pretrained_model=None)
elif self.n_layers == 101:
self.link = resnet.ResNet101Layers(pretrained_model=None)
elif self.n_layers == 152:
self.link = resnet.ResNet152Layers(pretrained_model=None)
def setUp(self):
self._config_user = chainer.using_config('dtype', self.dtype)
self._config_user.__enter__()
if self.n_layers == 50:
self.link = resnet.ResNet50Layers(pretrained_model=None,
downsample_fb=self.downsample_fb)
elif self.n_layers == 101:
self.link = resnet.ResNet101Layers(
pretrained_model=None, downsample_fb=self.downsample_fb)
elif self.n_layers == 152:
self.link = resnet.ResNet152Layers(
pretrained_model=None, downsample_fb=self.downsample_fb)