wheatdetection icon indicating copy to clipboard operation
wheatdetection copied to clipboard

Published 20 hours ago • verified publisher icon wuxinwang1997

How to train with mutil gpu?

Open thunderbird0902 opened this issue 4 years ago • 1 comments

I run your training code,and add torch.nn.DataParallel(model) to fitter func,but it didn't work.

thunderbird0902 avatar Jul 18 '20 15:07 thunderbird0902

Sorry, I use free Colab to train my model and I did not have a multi-gpu computer.

wuxinwang1997 avatar Jul 18 '20 15:07 wuxinwang1997