mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
ddp pickle
This commit is contained in:
1 parent
4aac5568a6
commit
d5dff384eb
1 file changed
+7
-5
@@ -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}',
|
||||
|
||||
Reference in new issue
Block a user