Update test_compressed_descriptor_se_atten.py

deepmodeling · Nov 13, 2024 · 9003b32 · 9003b32
1 parent 02b75b3
commit 9003b32
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/source/tests/pt/model/test_compressed_descriptor_se_atten.py b/source/tests/pt/model/test_compressed_descriptor_se_atten.py
@@ -115,11 +115,6 @@ def test_compressed_forward(self):
             self.box,
         )
 
-        if self.dtype == "float32":
-            result_pt = result_pt.to(torch.float32)
-        elif self.dtype == "float64":
-            result_pt = result_pt.to(torch.float64)
-
         self.se_atten.enable_compression(0.5)
         result_pt_compressed = eval_pt_descriptor(
             self.se_atten,