search

plemeri / InSPyReNet

Official PyTorch implementation of Revisiting Image Pyramid Structure for High Resolution Salient Object Detection (ACCV 2022)
MIT License
321 stars 61 forks source link

multi-gpu support for latest pytorch #4

Closed mrgono closed 1 year ago

mrgono commented 1 year ago

I found that the current training script does not support the latest pytorch DDP, so I modified the code and verify that my modification works for the latest pytorch DDP. Also, I check the memory leak in the pre-trained checkpoint load and solved the problem.

plemeri commented 1 year ago

Thank you very much! I'll review the code and see if there any conflicts with the original script.

mrgono commented 1 year ago

Glad it helped. Thank you for your fast response.