julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: remove deprecated Trainer.accelerator_backend

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-30 16:03:45 +02:00
parent 7b9b767113
commit 09e3ef1d0e
No known key found for this signature in database
GPG Key ID: 3331B0F18B6D4D93
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/abstract.py
View File

@ -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)