I am getting this error when I run the following code:
score_flattened = score_.view(B * NP * SQ, B2 * NS * SQ)
target_flattened = target_.view(B * NP * SQ, B2 * NS * SQ)
target_flattened = target_flattened.argmax(dim=1)

This error only happens on Google Colab but not on my own laptop (which has windows 10 on it)