From d5dff384eb5a2b885922065ec33c39ff255cf485 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 27 Apr 2020 07:07:03 -0400 Subject: [PATCH] ddp pickle --- pytorch_lightning/trainer/trainer.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) 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}',