Merge pull request #61 from breezedeus/pytorch

fix device value
breezedeus · Jul 2, 2023 · 259e8ba · 259e8ba
2 parents 40dd181 + 963ba92
commit 259e8ba
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -21,7 +21,7 @@ package:
 	rm -rf build
 	python setup.py sdist bdist_wheel
 
-VERSION = 1.2.3.1
+VERSION = 1.2.3.2
 upload:
 	python -m twine upload  dist/cnstd-$(VERSION)* --verbose
 

diff --git a/RELEASE.md b/RELEASE.md
@@ -1,5 +1,10 @@
 # Release Notes
 
+# Update 2023.07.02：发布 V1.2.3.2
+
+主要变更：
+* 修复参数 `device` 的取值bug，感谢 @Shadow-Alex 。
+
 # Update 2023.06.30：发布 V1.2.3.1
 
 主要变更：

diff --git a/cnstd/__version__.py b/cnstd/__version__.py
@@ -17,4 +17,4 @@
 # specific language governing permissions and limitations
 # under the License.
 
-__version__ = '1.2.3.1'
+__version__ = '1.2.3.2'
diff --git a/cnstd/yolov7/layout_analyzer.py b/cnstd/yolov7/layout_analyzer.py
@@ -155,6 +155,8 @@ def __init__(
         self._model_type = model_type
         self._model_backend = model_backend
 
+        if device.lower().strip() in ('cuda', 'cuda:0', 'gpu'):
+            device = '0'
         self.device = select_device(device)
 
         self._assert_and_prepare_model_files(model_fp, root)