Skip to content

Commit

Permalink
rebased with latest changes
Browse files Browse the repository at this point in the history
  • Loading branch information
lezwon committed Jun 20, 2020
1 parent 0a2c2e3 commit a3269e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -360,7 +360,7 @@ def __init__(

if tpu_cores is None:
tpu_cores = num_tpu_cores
self.tpu_cores = parse_tpu_cores(tpu_cores)
self.tpu_cores = _parse_tpu_cores(tpu_cores)
self.on_tpu = self.tpu_cores is not None

self.tpu_id = self.tpu_cores[0] if isinstance(self.tpu_cores, list) else None
Expand Down

0 comments on commit a3269e9

Please sign in to comment.