From 4032446a8f5fd4311c585c95c862a9253dd2a11e Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Mon, 27 Jul 2020 22:56:25 +0200 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Ethan Harris --- pytorch_lightning/accelerator_backends/tpu_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/accelerator_backends/tpu_backend.py b/pytorch_lightning/accelerator_backends/tpu_backend.py index fac8594f0358fa..95203550eb37e1 100644 --- a/pytorch_lightning/accelerator_backends/tpu_backend.py +++ b/pytorch_lightning/accelerator_backends/tpu_backend.py @@ -118,7 +118,7 @@ def tpu_train_in_process(self, tpu_core_idx: int, model: LightningModule, traine # persist info in spawn trainer.transfer_distrib_spawn_state_on_fit_end(model, mp_queue, results) - def __save_end_of_training_weights(self, model: LightningModule, trainer=None): + def __save_end_of_training_weights(self, model: LightningModule, trainer): # when training ends on these platforms dump weights to get out of the main process if trainer.on_colab_kaggle: rank_zero_warn('cleaning up... please do not interrupt')