You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
self.queue[ind, :, ptr:ptr + batch_size] = keys.T
RuntimeError: The expanded size of the tensor (6) must match the existing size (16) at non-singleton dimension 1. Target sizes: [128, 6]. Tensor sizes: [128, 16]
The text was updated successfully, but these errors were encountered:
It looks like ptr will begin at 506 or other numbers when ContrastiveTrainer do the forward, it will leads to when tload the last batch data it will out of queue range
Have you met this problem?
self.queue[ind, :, ptr:ptr + batch_size] = keys.T
RuntimeError: The expanded size of the tensor (6) must match the existing size (16) at non-singleton dimension 1. Target sizes: [128, 6]. Tensor sizes: [128, 16]
The text was updated successfully, but these errors were encountered: