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