Update teardown for TPU acc (#7211)

This commit is contained in:
Kaushik B 2021-04-26 18:00:46 +05:30 committed by GitHub
parent bc3f08b0e3
commit 6be0a859db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -50,6 +50,9 @@ class TPUAccelerator(Accelerator):
raise MisconfigurationException("TPUs only support a single tpu core or tpu spawn training.")
return super().setup(trainer, model)
def teardown(self) -> None:
pass
def run_optimizer_step(
self, optimizer: Optimizer, optimizer_idx: int, lambda_closure: Callable, **kwargs: Any
) -> None: