qubvel-org/segmentation_models.pytorch

Add multi-gpu training example script

オープン

#913 opened on 2024/08/23

 (0 件のコメント) (0 件のリアクション) (0 人の担当者)Python (1,842 件のフォーク)github user discovery
examplesgood first issue

Repository metrics

Stars
 (11,704 個のスター)
PR merge metrics
 (平均マージ 4h 14m) (30d で 24 merged PRs)

説明

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:

コントリビューターガイド