torchstat icon indicating copy to clipboard operation
torchstat copied to clipboard

Published 20 hours ago • verified publisher icon Swall0w

Spiking Neural Network support?

Open MrPeterJin opened this issue 3 years ago • 0 comments

Hello! Thanks for your great work at first.

I want to check my Spiking Neural Network model's GPU using torchstat, but it seems not to support inputs with SNN layers... Do you have any future plans to integrate SNN layers into your packages?

MrPeterJin avatar Dec 09 '21 02:12 MrPeterJin