Complete xlayers.rearrange

2021-06-08 23:47:52 -07:00
parent f9bbf974de
commit 744ce97bc5
10 changed files with 218 additions and 96 deletions
--- a/tests/test_super_att.py
+++ b/tests/test_super_att.py
@@ -29,6 +29,7 @@ class TestSuperSelfAttention(unittest.TestCase):
        abstract_child = abstract_space.random(reuse_last=True)
        print("The abstract child program is:\n{:}".format(abstract_child))
        model.set_super_run_type(super_core.SuperRunMode.Candidate)
+        model.enable_candidate()
        model.apply_candidate(abstract_child)
        outputs = model(inputs)
        return abstract_child, outputs