lt-tensor 0.0.1a28__tar.gz → 0.0.1a29__tar.gz

{lt_tensor-0.0.1a28 → lt_tensor-0.0.1a29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a28
+Version: 0.0.1a29
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336

{lt_tensor-0.0.1a28 → lt_tensor-0.0.1a29}/lt_tensor/model_base.py

@@ -137,16 +137,24 @@ class _Devices_Base(nn.Module):
             )
 
     def _apply_device(self):
-        """Add here components that are needed to have device applied to them,
-        that usually the '.to()' function fails to apply
-
-        example:
-        ```
-        def _apply_device_to(self):
-            self.my_tensor = self.my_tensor.to(device=self.device)
-        ```
-        """
-        pass
+        """This may be seem as overkill, but its necessary"""
+        for modules in self.modules():
+            try:
+                modules.to(self.device)
+            except:
+                pass
+
+        for buffer in self.buffers():
+            try:
+                buffer.to(self.device)
+            except:
+                pass
+
+        for tensor in self.parameters():
+            try:
+                tensor.to(self.device)
+            except:
+                pass
 
     def _to_dvc(
         self, device_name: str, device_id: Optional[Union[int, torch.device]] = None

{lt_tensor-0.0.1a28 → lt_tensor-0.0.1a29}/lt_tensor/processors/audio.py

@@ -119,14 +119,6 @@ class AudioProcessor(Model):
             (torch.hann_window(self.cfg.win_length) if window is None else window),
         )
 
-    def _apply_device(self):
-        self._mel_spec.to(device=self.device)
-        self.mel_rscale.to(device=self.device)
-        try:
-            self.window.to(device=self.device)
-        except:
-            pass
-
     def from_numpy(
         self,
         array: np.ndarray,

{lt_tensor-0.0.1a28 → lt_tensor-0.0.1a29}/lt_tensor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1a28
+Version: 0.0.1a29
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336

{lt_tensor-0.0.1a28 → lt_tensor-0.0.1a29}/setup.py

@@ -4,7 +4,7 @@ with open("README.md", "r", encoding="utf-8") as f:
     long_description = f.read()
 
 setup(
-    version="0.0.1a28",
+    version="0.0.1a29",
     name="lt-tensor",
     description="General utilities for PyTorch and others. Built for general use.",
     long_description=long_description,