Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ddp pickle

This commit is contained in:
William Falcon 2020-04-27 07:08:34 -04:00
parent d5dff384eb
commit d2989f76e6
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pytorch_lightning/trainer/trainer.py
View File

@ -614,19 +614,8 @@ class Trainer(
return bool(parsing.strtobool(x)) return bool(parsing.strtobool(x))
if arg == 'gpus': if arg == 'gpus':
def allowed_type(x): allowed_type = Trainer.allowed_type
if ',' in x: arg_default = Trainer.arg_default
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( parser.add_argument(
f'--{arg}', f'--{arg}',
@ -639,6 +628,18 @@ class Trainer(
return parser return parser
def allowed_type(x):
if ',' in x:
return str(x)
else:
return int(x)
def arg_default(x):
if ',' in x:
return str(x)
else:
return int(x)
@classmethod @classmethod
def from_argparse_args(cls, args, **kwargs): def from_argparse_args(cls, args, **kwargs):