Hebbian rule throws an error

[marina-kaz]

Hi!

I find this repository very curious, thank you!

However, I've tried to run the code with the Hebbian update rule, and I run across several exceptions.

1. It seems that in pytorch_hebbian/handlers/tensorboard_logger.py in line 8, BaseWeightsHandler should be imported instead of BaseWeightsHistHandler.

2. Running HebbsRule from hebb.py results with RuntimeError: 1D tensors expected, but got 2D and 1D tensors. I believe it is due to the torch.dot(w, x) operation: in PyTorch, dot only supports computing the dot product of two 1D tensors with the same number of elements, in contrast to NumPy.

Is there a way to fix it? Do you have any plans on continuing working on this project?

[julestalloen]

Hi @marina-kaz

What version of PyTorch are you using? I did not run into these issues so I'm assuming it is because of a version mismatch.

I am no longer maintaining this repository but updates and contributions are welcome.