SRN-Deblur icon indicating copy to clipboard operation
SRN-Deblur copied to clipboard

puzzle on training speed

Open ChristianoPenaldo opened this issue 6 years ago • 2 comments

I train the model on a Tesla M40 GPU. However, I could only train 40 steps within 1 minute. I admit that Titan Xp is faster than M40, so is multi-GPU training feasible?

ChristianoPenaldo avatar Jan 29 '19 18:01 ChristianoPenaldo

Yes. I think multi-gpu is possible. We use only common modules, which should be easy to put on multi-gpu. But the current codes need to be modified to use multi-gpu.

jiangsutx avatar Feb 20 '19 09:02 jiangsutx

How to modify the code to meet multi-gpu training

chenxinwhut avatar Sep 21 '22 02:09 chenxinwhut