IsoNet-cryoET / IsoNet

Self-supervised learning for isotropic cryoET reconstruction
https://www.nature.com/articles/s41467-022-33957-8
MIT License
70 stars 13 forks source link

Update train.py #36

Closed LianghaoZhao closed 1 year ago

LianghaoZhao commented 1 year ago

fix bug in multi-GPU training