qubvel-org/segmentation_models.pytorch

Add multi-gpu training example script

Open

#913 opened on Aug 23, 2024

View on GitHub
 (0 comments) (0 reactions) (0 assignees)Python (1,842 forks)github user discovery
examplesgood first issue

Repository metrics

Stars
 (11,704 stars)
PR merge metrics
 (Avg merge 4h 14m) (24 merged PRs in 30d)

Description

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:

Contributor guide

Research direction
Create a Python script or Jupyter notebook that demonstrates multi GPU fine tuning of a segmentation model using PyTorch Lightning or plain PyTorch. Include correct metric gathering with torchmetrics and result visualization.
Tech stack
python
Domain
backenddocumentation
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
Half day
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Fresh
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
PythonPyTorchPyTorch Lightning
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
70