Skip to content

Commit 3a44c73

Browse files
committed
indentation
1 parent 53338f8 commit 3a44c73

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

pytorch_lightning/plugins/training_type/tpu_spawn.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
class TPUSpawnPlugin(DDPSpawnPlugin):
2626

2727
def __init__(self, parallel_devices: Sequence[int], num_nodes: int = 1, **kwargs: Dict[str, Any]) -> None:
28-
2928
super().__init__(
3029
parallel_devices, num_nodes=num_nodes, cluster_environment=None, sync_batchnorm=False, **kwargs
3130
)
@@ -179,15 +178,13 @@ def __load_weights_on_main_process(self) -> None:
179178

180179
self.lightning_module = model
181180

182-
@property
183-
def xmp_spawn_kwargs(self):
181+
@property
182+
def xmp_spawn_kwargs(self):
184183
return {
185184
"args": (self.lightning_module, trainer, self.mp_queue),
186185
"nproc": len(self.parallel_devices),
187186
"start_method": self.start_method
188187
}
189-
190-
191188

192189
def start_training(self, trainer) -> None:
193190
xmp.spawn(

0 commit comments

Comments
 (0)