I\'m using rtx 3080, python 3.7, torch 1.7.1, torchvision 0.8.2
i tried chaning python version and batch size etc. But doesn\'t work..
what can i do to fix this pr
