fix: remove deprecated Trainer.accelerator_backend

2021-06-30 16:03:45 +02:00
parent 7b9b767113
commit 09e3ef1d0e
1 changed files with 1 additions and 1 deletions
--- a/prototorch/models/abstract.py
+++ b/prototorch/models/abstract.py
@@ -73,7 +73,7 @@ class PrototypeModel(ProtoTorchBolt):

    @final
    def reconfigure_optimizers(self):
-        self.trainer.accelerator_backend.setup_optimizers(self.trainer)
+        self.trainer.accelerator.setup_optimizers(self.trainer)

    def add_prototypes(self, *args, **kwargs):
        self.proto_layer.add_components(*args, **kwargs)