def test_save_model_to_tiledb_array_subclassed(self):
# Subclassed
self.subclassed_model = testing_utils.get_small_subclass_mlp(
num_hidden=1, num_classes=2)
tiledb_uri = os.path.join(self.get_temp_dir(), "model_array")
tiledb_model_obj = TensorflowTileDB(uri=tiledb_uri)
with self.assertRaises(NotImplementedError):
tiledb_model_obj.save(model=self.subclassed_model,
include_optimizer=False)
def setUp(self): super(TestSaveModel, self).setUp() self.model = testing_utils.get_small_sequential_mlp(1, 2, 3) self.subclassed_model = testing_utils.get_small_subclass_mlp(1, 2)
def setUp(self): self.model = testing_utils.get_small_sequential_mlp(1, 2, 3) self.subclassed_model = testing_utils.get_small_subclass_mlp(1, 2)
def setUp(self):
self.model = testing_utils.get_small_sequential_mlp(1, 2, 3)
self.subclassed_model = testing_utils.get_small_subclass_mlp(1, 2)
