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

5 RTX4090s are still not enough for this project? #158

Open
kawaiitakagi opened this issue Aug 10, 2024 · 1 comment
Open

5 RTX4090s are still not enough for this project? #158

kawaiitakagi opened this issue Aug 10, 2024 · 1 comment

Comments

@kawaiitakagi
Copy link

I only used the co_dino_5scale_swin_large_3x_coco.py config which is smaller, and has resized my pictures to 512*512, with 5 RTX4090 GPUs to train. However the memory was still not enough. Is that normal?
Do the config files in this project support mixed precision training? I have added this line into the config trying to reduce memory consumed, but it seemed no use: fp16 = dict(loss_scale=512.).

@TempleX98
Copy link
Collaborator

  1. Please show me your training config.
  2. This repo does not support fp16 training.

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