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

Sync BatchNorm costs lots of time and cause code unworkable #17

Open
ppbangKGT opened this issue Jan 12, 2024 · 1 comment
Open

Sync BatchNorm costs lots of time and cause code unworkable #17

ppbangKGT opened this issue Jan 12, 2024 · 1 comment

Comments

@ppbangKGT
Copy link

Hi, thanks for your nice work!
When I run your code, there met a problem.
Because my device is unable to connect to the internet, I download the resnet50 pth pretrained on ImageNet instead of using the mm-pretrain model. This means I change the encoder config init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50') to init_cfg=dict(type='Pretrained', checkpoint='/data1/pbw/LATR-main/resnet50-19c8e357.pth')
By doing so, I successfully run the code. But it stops at operation "Convert model with Sync BatchNorm", and cannot go on. I have taken a screenshot as follows.
1705052282934
Could you give me some advice? Thanks for your time.

@zjulabwjt
Copy link

Hi, thanks for your nice work! When I run your code, there met a problem. Because my device is unable to connect to the internet, I download the resnet50 pth pretrained on ImageNet instead of using the mm-pretrain model. This means I change the encoder config init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50') to init_cfg=dict(type='Pretrained', checkpoint='/data1/pbw/LATR-main/resnet50-19c8e357.pth') By doing so, I successfully run the code. But it stops at operation "Convert model with Sync BatchNorm", and cannot go on. I have taken a screenshot as follows. 1705052282934 Could you give me some advice? Thanks for your time.

have you solved this problem?

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