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.")