acids-ircam / RAVE

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder
Other
1.35k stars 184 forks source link

receptive field computation for multichannel models #292

Open victor-shepardson opened 9 months ago

victor-shepardson commented 9 months ago

I think there is a bug when computing the receptive field for multichannel models: https://github.com/acids-ircam/RAVE/blob/8b250310fecfe61a6d9d53e8e5551851f4638d35/rave/core.py#L200

should the grad tensor be indexed to channel zero like grad = x.grad[0,0].reshape(-1) ?