Merge pull request #16 from cy-xu/fix-tensor-deprecation

Fixed pytorch deprecation tensor type.

Author: cmccully

cosmic_conn/dl_framework/cosmic_conn.py

@@ -60,7 +60,6 @@ def initialize(self, opt):
         self.build_models(opt)
 
     def build_models(self, opt):
-        torch.set_default_tensor_type(torch.FloatTensor)
         torch.set_default_dtype(torch.float32)
 
         # group norm parameters

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cosmic_conn"
-version = "0.6.0"
+version = "0.6.1"
 description = "Cosmic-CoNN: A Cosmic Ray Detection Deep Learning Framework, Dataset, and Toolkit"
 readme = "README.md"
 license = { text = "GNU General Public License v3" }