search

genn-team / ml_genn

A library for deep learning with Spiking Neural Networks (SNN).
https://ml-genn.readthedocs.io
GNU Lesser General Public License v2.1
24 stars 7 forks source link

AvgPoolConv2D does not work with weight initialisers #119

Open neworderofjamie opened 4 days ago

neworderofjamie commented 4 days ago

as it tries to calculate:

scaled_weight = self.weight.flatten() / (pool_kh * pool_kw)

which only works if weight is a numpy array