qubvel-org/segmentation_models.pytorch

Add multi-gpu training example script

开放

#913 创建于 2024年8月23日

 (0 条评论) (0 个反应) (0 位负责人)Python (1,842 个派生)github user discovery
examplesgood first issue

仓库指标

星标
 (11,704 个星标)
PR 合并指标
 (平均合并 4小时 14分钟) (30 天内合并 24 个 PR)

描述

It would be great to have more examples for fine-tuning in the library!

Similar to current examples for binary-segmentaion and multi-label segmentation would be great to have Multi-GPU training example. This can be handled with PyTorch-Lightning or with pure Pytorch.

It might be a .py script or Jupyter notebook. The example should showcase how to

  • fine-tune a model with pytorch-lightning (or any other fine-tuning framework, even a plain pytorch)
  • compute metrics with correct gathering of results (maybe torchmetrics can be utilized)
  • visualize results

In case anyone wants to work on this you are welcome! Just notify in this issue, and then ping me to review a PR when it's ready.

Fixes:

贡献者指南