Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard
verified publisher icon Sense-X

5 RTX4090s are still not enough for this project?

Open kawaiitakagi opened this issue 1 year ago • 1 comments

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.).

kawaiitakagi avatar Aug 10 '24 07:08 kawaiitakagi

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

TempleX98 avatar Aug 11 '24 08:08 TempleX98