Add example for dynamic components in callbacks

2021-05-31 11:39:24 +02:00
parent db064b5af1
commit 2a218c0ede
2 changed files with 28 additions and 21 deletions
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -137,7 +137,7 @@ class GLVQ(AbstractPrototypeModel):

    def increase_prototypes(self, initializer, distribution):
        self.proto_layer.increase_components(initializer, distribution)
-        #self.trainer.accelerated_backend.setup_optimizers(self)
+        self.trainer.accelerator_backend.setup_optimizers(self.trainer)

    def __repr__(self):
        super_repr = super().__repr__()