v1.8.0

Author: TylerYep

.gitignore

@@ -5,6 +5,7 @@ __pycache__/
 .vscode/
 .DS_Store
 profile.txt
+Archive.zip
 
 # C extensions
 *.so

Archive.zip

@@ -184,7 +184,9 @@ Args:
 
     device (torch.Device):
             Uses this torch device for model and input_data.
-            If not specified, uses result of torch.cuda.is_available().
+            If not specified, uses the dtype of input_data if given, or the
+            parameters of the model. Otherwise, uses the result of
+            torch.cuda.is_available().
             Default: None
 
     dtypes (List[torch.dtype]):

README.md

@@ -11,4 +11,4 @@
     "Units",
     "Verbosity",
 )
-__version__ = "1.7.2"
+__version__ = "1.8.0"

torchinfo/__init__.py

@@ -140,7 +140,9 @@ class name as the key. If the forward pass is an expensive operation,
 
         device (torch.Device):
                 Uses this torch device for model and input_data.
-                If not specified, uses result of torch.cuda.is_available().
+                If not specified, uses the dtype of input_data if given, or the
+                parameters of the model. Otherwise, uses the result of
+                torch.cuda.is_available().
                 Default: None
 
         dtypes (List[torch.dtype]):