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)