Example #1
0
 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)
Example #2
0
    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)