Update teardown for TPU acc (#7211)
This commit is contained in:
parent
bc3f08b0e3
commit
6be0a859db
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue