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.39k stars 137 forks source link

Update utils.py #117

Closed shinbehavior closed 11 months ago

shinbehavior commented 11 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.