iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix test bugs

Browse Source
This commit is contained in:
D-X-Y
2021-05-06 16:43:31 +08:00
parent 4c14c7b85b
commit f6a024a6ff
4 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_super_norm.py
View File

@@ -64,7 +64,7 @@ class TestSuperSimpleNorm(unittest.TestCase):
model.apply_verbose(True)
print(model.super_run_type)
self.assertTrue(model[1].bias)
self.assertTrue(model[2].bias)
inputs = torch.rand(20, 10)
print("Input shape: {:}".format(inputs.shape))
@@ -80,6 +80,6 @@ class TestSuperSimpleNorm(unittest.TestCase):
model.set_super_run_type(super_core.SuperRunMode.Candidate)
model.apply_candidate(abstract_child)
output_shape = (20, abstract_child["1"]["_out_features"].value)
output_shape = (20, abstract_child["2"]["_out_features"].value)
outputs = model(inputs)
self.assertEqual(tuple(outputs.shape), output_shape)