Bug in the code

[phquanta]

I've started training and i've noticed that after reloading model checkpoint i'm seeing mismatch for W_i vectors with hardcoded dime1=32 parameter in save_disc branch and dim1=num_videos in master branch. None of them seems to be correct, looks like it should be batch_size because we are loading W_i at each training iteration.

[hanweikung]

I had the same problem. Thank you for the suggestion