Distributed training in CellViT

TIO-IKIM / CellViT

CellViT: Vision Transformers for Precise Cell Segmentation and Classification

https://doi.org/10.1016/j.media.2024.103143

Other

189 stars 27 forks source link

Closed CYYJL closed 3 months ago

CYYJL commented 3 months ago

Hi, if I want to conduct distributed training, how can I implement it in your code?

FabianHoerst commented 3 months ago

We currently have no distributed training implemented. If you are interested in doing so, I would advice you to follow the official PyTorch Tutorial: https://pytorch.org/tutorials/intermediate/ddp_tutorial.html?utm_source=distr_landing&utm_medium=intermediate_ddp_tutorial