From a22a8142ac65668781a6e6f76d3c4e55ea7c249a Mon Sep 17 00:00:00 2001 From: Krishna Penukonda Date: Sat, 18 Apr 2020 06:18:29 +0800 Subject: [PATCH] Allow Trainer's `gpus` arg type to be subclass of currently accepted types (#1423) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Fixed Trainer `gpus` arg type issue Fixes #1388 * Disallow boolean gpus parameter Co-Authored-By: Adrian Wälchli * Fixed missing paranthesis Co-authored-by: Adrian Wälchli --- pytorch_lightning/trainer/distrib_parts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/distrib_parts.py b/pytorch_lightning/trainer/distrib_parts.py index fb47ad21..5e26d499 100644 --- a/pytorch_lightning/trainer/distrib_parts.py +++ b/pytorch_lightning/trainer/distrib_parts.py @@ -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.")