search

MCG-NJU / VideoMAE

[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
https://arxiv.org/abs/2203.12602
Other
1.32k stars 131 forks source link

Update utils.py #117

Closed hagonata closed 8 months ago

hagonata commented 9 months ago

import torch._six to import torch

So you can use it with all torch versions. Or put somewhere about <2.0 torch version.