search

miladmozafari / SpykeTorch

High-speed simulator of convolutional spiking neural networks with at most one spike per neuron.
GNU General Public License v3.0
382 stars 101 forks source link

Simulation Time #2

Closed hpnair closed 3 years ago

hpnair commented 5 years ago

Hi Milad,

How much time did your simulations take for the Deep network to run on MNIST on CPUs vs GPUs?

miladmozafari commented 5 years ago

Hi, For GPUs, we have reported the execution time in SpykeTorch’s paper (https://www.frontiersin.org/articles/10.3389/fnins.2019.00625/full). I have never tried deep net scripts on CPUs, however I do not recommend running them on CPUs. But if you want to have an estimation, you can train each layer just for one epoch and do the math to find the execution time of the actual simulation.

hpnair commented 5 years ago

Hi,

Thanks for your prompt reply. Appreciate it!