Minor changes in order to get the code running, also added time tracking for...

Minor changes in order to get the code running, also added time tracking for each iteration in train_moving_mnist.py