Updated ce_max_membrane_loss() documentation

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

https://snntorch.readthedocs.io/en/latest/

MIT License

1.35k stars 222 forks source link

Closed vinniesun closed 1 year ago

vinniesun commented 1 year ago

Updated snntorch.functional.ce_max_membrane_loss to be clearer with how the loss function expects its inputs