Skip to content

GitLab

"launch/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "5d89a0c858b65599922a89967d415ff9010a76f4"
Commit 3de188e9 authored by Christina Floristean's avatar Christina Floristean
Browse files

Minor model test fix

parent a18f98cf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_model.py
View file @ 3de188e9
...@@ -55,7 +55,7 @@ class TestModel(unittest.TestCase): ...@@ -55,7 +55,7 @@ class TestModel(unittest.TestCase):
n_res = consts.n_res n_res = consts.n_res
n_extra_seq = consts.n_extra n_extra_seq = consts.n_extra
c = model_config(consts.model, train=True) c = model_config(consts.model)
c.model.evoformer_stack.no_blocks = 4 # no need to go overboard here c.model.evoformer_stack.no_blocks = 4 # no need to go overboard here
c.model.evoformer_stack.blocks_per_ckpt = None # don't want to set up c.model.evoformer_stack.blocks_per_ckpt = None # don't want to set up
# deepspeed for this test # deepspeed for this test
......
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