mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-25 12:50:57 +08:00
updated args
This commit is contained in:
1 parent
684dfd0a38
commit
bf3b86ce4d
1 file changed
+1
-1
@@ -95,7 +95,7 @@ def main(hparams, cluster, results_dict):
|
||||
# gpus are ; separated for inside a node and , within nodes
|
||||
gpu_list = None
|
||||
if hparams.gpus is not None:
|
||||
gpu_list = map(int, hparams.gpus.split(';'))
|
||||
gpu_list = [int(x) for x in hparams.gpus.split(';')]
|
||||
|
||||
# configure trainer
|
||||
trainer = Trainer(
|
||||
|
||||
Reference in new issue
Block a user