v1.8.0

TylerYep · May 14, 2023 · 73ed568 · 73ed568
1 parent c4f43aa
commit 73ed568
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@ __pycache__/
 .vscode/
 .DS_Store
 profile.txt
+Archive.zip
 
 # C extensions
 *.so

diff --git a/Archive.zip b/Archive.zip
diff --git a/README.md b/README.md
@@ -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]):

diff --git a/torchinfo/__init__.py b/torchinfo/__init__.py
@@ -11,4 +11,4 @@
     "Units",
     "Verbosity",
 )
-__version__ = "1.7.2"
+__version__ = "1.8.0"
diff --git a/torchinfo/torchinfo.py b/torchinfo/torchinfo.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]):