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)