qubvel-org/segmentation_models.pytorch

Add multi-gpu training example script

Aperta

#913 aperta il 23 ago 2024

Vedi su GitHub
 (0 commenti) (0 reazioni) (0 assegnatari)Python (1842 fork)github user discovery
examplesgood first issue

Metriche repository

Star
 (11.704 stelle)
Metriche merge PR
 (Merge medio 4h 14m) (24 PR mergiate in 30 g)

Descrizione

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:

Guida contributor

Direzione di ricerca
Crea uno script Python o un Jupyter notebook che dimostri il fine tuning multi GPU di un modello di segmentazione utilizzando PyTorch Lightning o PyTorch puro. Includi la corretta raccolta delle metriche con torchmetrics e la visualizzazione dei risultati.
Tech stack
python
Dominio
backenddocumentation
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorchPyTorch Lightning
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
70