Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training time for kinetics-400 #96

Closed
AbdelrahmanShakerYousef opened this issue Nov 16, 2022 · 6 comments
Closed

Training time for kinetics-400 #96

AbdelrahmanShakerYousef opened this issue Nov 16, 2022 · 6 comments

Comments

@AbdelrahmanShakerYousef

Hello,

Thank you for sharing the codebase of your exciting work.

Could you please let me know the training time for pertaining & training on kinetics-400 and the resources you used?

Thank you!

@Andy1621
Copy link
Collaborator

Thanks for your question.
In my original experiments, all the models are run on a single V100 machine (8GPU) except the largest model.
For detailed training time, you can find the provided logs! It may take ~2 days to train K400 for 110 epochs with 16 frames.

For your experiments, you can use fewer epochs (~55) and fewer frames (~8) to verify your idea fastly.

@AbdelrahmanShakerYousef
Copy link
Author

@Andy1621
Copy link
Collaborator

It's a good question.
As a new backbone, it needs to be pre-trained on ImageNet first, which will cost 12 day. Thus, it actually needs about 34 days to verify it on K400.
However, the performance trend is similar. Those models work better on ImageNet often work better on K400.

@AbdelrahmanShakerYousef
Copy link
Author

Thank you Andy for your reply. Is it 12 days or 34 days? I guess your text is unclear about that.

@Andy1621
Copy link
Collaborator

Andy1621 commented Nov 18, 2022

1 to 2 days and 3 to 4 days.

The ~ automatically changes to delete.😊

@Andy1621
Copy link
Collaborator

Hi! We have released UniFormerV2, which adopts pretrained ViT. It can save a lot of training resources. Welcome to try it!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants