I am training a Video Classification Model (from transformers import VideoMAEImageProcessor, VideoMAEForVideoClassification). It is working correctly in training phrase, but in validation phrase, it only use cpu for validation, no gpu is used. Below is the link to notebook. Thanks for helping!
notebook