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)
def setUp(self): self.link = resnet.ResNet50Layers(pretrained_model=None)