topoloss 0.0.1

topoloss

topoloss

Induce brain-like topographic structure in your neural networks

pip install topoloss

Example

import torchvision.models as models
from topoloss import TopoLoss, LaplacianPyramid

model = models.resnet18(weights = "DEFAULT")

topo_loss = TopoLoss(
    losses = [
        LaplacianPyramid(layer_name = 'fc',factor_h=3.0, factor_w=3.0),
    ],
)

print(topo_loss.compute(model=model, reduce_mean = True)) ## returns a single number as tensor for backward()
print(topo_loss.compute(model=model, reduce_mean = False)) ## returns a dict with layer names as keys

Running tests

pytest -vvx tests