Recurrent-Neural-Networks-from-scratch-using-PyTorch icon indicating copy to clipboard operation
Recurrent-Neural-Networks-from-scratch-using-PyTorch copied to clipboard

Run time statistics

Open Sherrylife opened this issue 2 months ago • 0 comments

Hi, @georgeyiasemis, it is nice to see you doing such a great job. I was wondering if you have compared your implementation and the official implementation of pytorch? I tested it and it seemed that your implementation of LSTM is 10x slower than the official implementation of LSTM using pytorch under the same input. Is there any way to improve the speed in your implementation?

Sherrylife avatar Apr 16 '24 08:04 Sherrylife