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

the training iteration is abnormally large #42

Closed
bolin-chen opened this issue Jan 4, 2022 · 2 comments
Closed

the training iteration is abnormally large #42

bolin-chen opened this issue Jan 4, 2022 · 2 comments

Comments

@bolin-chen
Copy link

I used 4 gpus(2080ti) to train swin_small with config swin_small_patch244_window877_kinetics400_1k.py. The dataset I used is hacs(50w videos). The following is some of my training log:

image

I find that the train iteration is abnormally large based on my config(dataset size(50w) and batch_size(8)), which leads to the long training time. Is that normal?

@MartinXM
Copy link

Check your optimizer_config, set update_interval=1 if your GPU memory is large enough, it would make your training faster.

@bolin-chen
Copy link
Author

Get it. It seems that the actual batch size is batch_size / update_interval.

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