ddp pickle

2020-04-27 07:07:03 -04:00 · 2020-04-27 07:07:03 -04:00 · d5dff384eb
parent 4aac5568a6
commit d5dff384eb
1 changed files with 7 additions and 5 deletions
--- 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}',