wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow Trainer's gpus arg type to be subclass of currently accepted types (#1423)

Browse Source 
* Fixed Trainer `gpus` arg type issue

Fixes #1388

* Disallow boolean gpus parameter

Co-Authored-By: Adrian Wälchli <aedu.waelchli@gmail.com>

* Fixed missing paranthesis

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
This commit is contained in:
Krishna Penukonda
2020-04-17 18:18:29 -04:00
committed by GitHub
co-authored by Adrian Wälchli
parent 1ee2837d62
commit a22a8142ac
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/distrib_parts.py
+1 -1
View File
@@ -566,7 +566,7 @@ def check_gpus_data_type(gpus):
:return: return unmodified gpus variable
"""
if gpus is not None and type(gpus) not in (int, str, list):
if gpus is not None and (not isinstance(gpus, (int, str, list)) or isinstance(gpus, bool)):
raise MisconfigurationException("GPUs must be int, string or list of ints or None.")