Add examples/dynamic_components.py

This commit is contained in:
Jensun Ravichandran
2021-05-31 00:32:27 +02:00
parent b7edee02c3
commit cd73f6c427
3 changed files with 59 additions and 1 deletions

View File

@@ -1,6 +1,5 @@
import pytorch_lightning as pl
import torch
from prototorch.functions.competitions import wtac
from torch.optim.lr_scheduler import ExponentialLR

View File

@@ -120,6 +120,9 @@ class GLVQ(AbstractPrototypeModel):
# def predict_step(self, batch, batch_idx, dataloader_idx=None):
# pass
def increase_prototypes(self, initializer, distribution):
self.proto_layer.increase_components(initializer, distribution)
def __repr__(self):
super_repr = super().__repr__()
return f"{super_repr}"