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