search

AllenCellModeling / pytorch_fnet

Three dimensional cross-modal image inference
Other
151 stars 61 forks source link

Can I train on CPUs? #166

Closed michalstepniewski closed 4 years ago

michalstepniewski commented 4 years ago

Can I train on CPUs?

gregjohnso commented 4 years ago

Hi @michalstepniewski, unfortunately I don't think the code is implemented to that right now. Even if we were to change the implementation, it would take 50x - 100x longer to train.