Skip to content

GitLab

"tools/cfgs/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "2e4d090e05198c74e9bb224274a63bbd12371192"
Commit f0fc7d91 authored by Jennifer's avatar Jennifer
Browse files

merging changes from main

parents ff368008 df4dfacb
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/test_deepspeed_evo_attention.py
View file @ f0fc7d91
...@@ -306,7 +306,6 @@ class TestDeepSpeedKernel(unittest.TestCase): ...@@ -306,7 +306,6 @@ class TestDeepSpeedKernel(unittest.TestCase):
batch["residx_atom37_to_atom14"] = batch[ batch["residx_atom37_to_atom14"] = batch[
"residx_atom37_to_atom14" "residx_atom37_to_atom14"
].long() ].long()
# print(batch["target_feat"].shape)
batch["target_feat"] = torch.nn.functional.one_hot(batch["aatype"], consts.msa_logits - 1).to(torch.float32) batch["target_feat"] = torch.nn.functional.one_hot(batch["aatype"], consts.msa_logits - 1).to(torch.float32)
batch["template_all_atom_mask"] = batch["template_all_atom_masks"] batch["template_all_atom_mask"] = batch["template_all_atom_masks"]
batch.update( batch.update(
......
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