Skip to content
Snippets Groups Projects
Commit 1e54e176 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[test.test_summary] Fix resunet test

parent 3603b3fd
No related branches found
No related tags found
1 merge request!13Cleanup model implementation
Pipeline #39946 passed
...@@ -42,7 +42,7 @@ class Tester(unittest.TestCase): ...@@ -42,7 +42,7 @@ class Tester(unittest.TestCase):
self.assertIsInstance(param, int) self.assertIsInstance(param, int)
def test_summary_resunet(self): def test_summary_resunet(self):
model = resunet() model = resunet50()
s, param = summary(model) s, param = summary(model)
self.assertIsInstance(s, str) self.assertIsInstance(s, str)
self.assertIsInstance(param, int) self.assertIsInstance(param, int)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment