Zhenyu Yang

Institute of Automation, Chinese Academy of Sciences

Results 3 comments of


                                            Zhenyu Yang

Training on multi GPUs are always OOM while it working well on single GPU

I have the same problem, and I can't run on muti gpus because the images run on different opus but the prompt runs on the single gpu. Do you have...

Training on multi GPUs are always OOM while it working well on single GPU

You can try setting the strategy='dp', then you can run muti opus, but you may get the same issue as mine.

How to Properly Reset Long Term Memory for Multiple Video Inferences Without Reloading the Model?

Same problem as you.