ToTensor is missing from kornia
USER@koeln:/netscratch/
USER/FR-template-pytorch$ /USER@koeln:/netscratch/
USER/FR-template-pytorch$ ./run_slurm_imagenet.sh
srun: job 102996 queued and waiting for resources
srun: job 102996 has been allocated resources
pyxis: creating container filesystem ...
pyxis: starting container ...
Traceback (most recent call last):
File "./src/multi-gpu.py", line 19, in
from model.resnet50 import PL_ResNet50
File "/netscratch/$USER/FR-template-pytorch/src/model/resnet50.py", line 10, in
from kornia.augmentation import RandomResizedCrop, RandomHorizontalFlip, CenterCrop, RandomCrop, Normalize, ToTensor
ImportError: cannot import name 'ToTensor'
srun: error: geneva: task 0: Exited with exit code 1
srun: Terminating job step 102996.0
USER@koeln:/netscratch/
USER/FR-template-pytorch$ grep kornia requirements.txt
kornia==0.4.0