Command for maskfeat pretraining

mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.

272 stars 34 forks source link

Command for maskfeat pretraining #12

Closed WHlTE-N0lSE closed 2 years ago

WHlTE-N0lSE commented 2 years ago

Request the pretraining command for maskfeat.This doesn't seem to be provided in the code. Thank you!

mx-mark commented 2 years ago

@WHlTE-N0lSE The pretraining repo of maskfeat may need to be optimized. Currently, the speed of data loader perform not well. Because the hog features extraction occupy most of time.

WHlTE-N0lSE commented 2 years ago

The speed is acceptable, but what is the shell command of maskfeat pretraining?

mx-mark commented 2 years ago

@WHlTE-N0lSE We have alleviated the speed problem in maskfeat. The shell command can be found here https://github.com/mx-mark/VideoTransformer-pytorch#training