mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
fixed demo
This commit is contained in:
1 parent
0c7fbc7178
commit
9f9d38673e
1 file changed
+1
-1
@@ -143,7 +143,7 @@ if __name__ == '__main__':
|
||||
|
||||
# TODO: make 1 param
|
||||
parent_parser.add_argument('--per_experiment_nb_gpus', type=int,
|
||||
help='how many gpus to use in a node')
|
||||
default=2, help='how many gpus to use in a node')
|
||||
parent_parser.add_argument('--gpus', type=str, default='-1',
|
||||
help='how many gpus to use in the node')
|
||||
|
||||
|
||||
Reference in new issue
Block a user