def test_works_with_mobilenet_v2(self,): mobilenet = backbone.MobileNetBackbone('mobilenet_v2') model = mtp.MTP(mobilenet, 3, 6, 2, input_shape=(3, 100, 100)) self._run(model)
def test_works_with_resnet_152(self,): rn_152 = backbone.ResNetBackbone('resnet152') model = mtp.MTP(rn_152, 3, 6, 2, input_shape=(3, 100, 100)) self._run(model)