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

Completely solve the problem of the number of steps in one epoch #101

Closed JerryFlymi closed 1 year ago

JerryFlymi commented 1 year ago

Make num_training_steps_per_epoch absolutely consistent with the calculations within the dataloader.

JerryFlymi commented 1 year ago

It may be equivalent to the previous realization.