search

mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.
272 stars 34 forks source link

While training viti I am getting thsi error #31

Open balajiiitg opened 1 year ago

balajiiitg commented 1 year ago

Predictions and targets are expected to have the same shape, but got torch.Size([8, 400]) and torch.Size([8]).