search

xuebinqin / DIS

This is the repo for our new project Highly Accurate Dichotomous Image Segmentation
Apache License 2.0
2.27k stars 270 forks source link

Multi gpu training #51

Open haruntrkmn opened 2 years ago

haruntrkmn commented 2 years ago

How can we perform multi-gpu training? Should we use DataParallel class of pytorch, or is there a parameter in the code for it?