Incorrect initialization of tracking in test

I might be wrong... but I'm pretty confident this line should be

pred = lbls[n_context-1]

If you initialize it at lbls[0] aren't you setting preds[0] to be lbls[0], while preds[1] is predicting lbls[n_context]?

https://github.com/ajabri/videowalk/blob/047f3f40135a4b1be2f837793b89c3dbfe7a6683/code/test.py#L147