search

tbepler / topaz

Pipeline for particle picking in cryo-electron microscopy images using convolutional neural networks trained from positive and unlabeled examples. Also featuring micrograph and tomogram denoising with DNNs.
GNU General Public License v3.0
169 stars 63 forks source link

Training on multi gpus #39

Closed nguyen14ck closed 4 years ago

nguyen14ck commented 4 years ago

Can we train the model on multi gpus? I set

--device=0,1

but that did not work.

Thanks

tbepler commented 4 years ago

No, we don't have support for multi GPU training. Unfortunately, this wouldn't provide any speed up for topaz train anwyay, because training isn't bottlenecked by the GPU operations.