Commit d76bc437 authored by patil-suraj's avatar patil-suraj
Browse files

add skeleton for model tests

parent 8c1f5197
...@@ -117,6 +117,24 @@ class ModelTesterMixin(unittest.TestCase): ...@@ -117,6 +117,24 @@ class ModelTesterMixin(unittest.TestCase):
image = model(*self.dummy_input) image = model(*self.dummy_input)
assert image is not None, "Make sure output is not None" assert image is not None, "Make sure output is not None"
def test_save_load(self):
pass
def test_determinism(self):
pass
def test_output(self):
pass
def test_forward_signature(self):
pass
def test_model_from_config(self):
pass
def test_training(self):
pass
class PipelineTesterMixin(unittest.TestCase): class PipelineTesterMixin(unittest.TestCase):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment