diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index d3f208810e..59cdf41b42 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -620,11 +620,13 @@ class Trainer( else: return int(x) - # def arg_default(x): - # if ',' in x: - # return str(x) - # else: - # return int(x) + def arg_default_fx(x): + if ',' in x: + return str(x) + else: + return int(x) + + arg_default = arg_default_fx parser.add_argument( f'--{arg}',