running CE loss does not work

[rabeehkarimimahabadi]

Hi I am running given example in README
thanks

Traceback (most recent call last):
File "main_ce.py", line 333, in
main()
File "main_ce.py", line 302, in main
loss, train_acc = train(train_loader, model, criterion, optimizer, epoch, opt)
File "main_ce.py", line 214, in train
acc1, acc5 = accuracy(output, labels, topk=(1, 5))
File "/remote/idiap.svm/user.active/rkarimi/dev/internship/SupContrast/util.py", line 48, in accuracy
correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

[Patrickctyyx]

try change correct_k = correct[:k].view(-1).float().sum(0, keepdim=True) to correct_k = correct[:k].reshape(-1).float().sum(0, keepdim=True) in utils.py line 48