Given that you already coded the model, you could for example follow https://stackoverflow.com/questions/7370801/measure-time-elapsed-in-python 8 to measure the training time (e.g. for one epoch) on both implementations.