Code

Add multi gpu support#3548

Code

Merged

HallerPatrick merged 9 commits intoflairNLP:masterflairNLP/flair:masterfrom jeffpicard:multi-gpujeffpicard/flair:multi-gpuCopy head branch name to clipboardNov 22, 2024

+314-25

Commits

Commits on Nov 15, 2024

• flairNLPGH-3429: Add experimental multi GPU support

  jeffpicard

  committed
  e71797c

  Copy full SHA for e71797c

• flairNLPGH-3429: Move process spawning inside .train; WIP sync gradients

  jeffpicard

  committed
  f777cc6

  Copy full SHA for f777cc6

• flairNLPGH-3429: Redirect forward_loss; User calls launch; Handle plugins

  jeffpicard

  committed
  c918093

  Copy full SHA for c918093

• fix transformers switching attention implementation

  jeffpicard

  committed
  0cbeb9f

  Copy full SHA for 0cbeb9f

• flairNLPGH-3429: Rename parameter group

  jeffpicard

  committed
  a590cd6

  Copy full SHA for a590cd6

• flairNLPGH-3429: Add example to docs; Guard against corpus being different on each process

  jeffpicard

  committed
  ffb8e29

  Copy full SHA for ffb8e29

• flairNLPGH-3429: Remove serializability requirement

  jeffpicard

  committed
  9a82b0e

  Copy full SHA for 9a82b0e

Commits on Nov 22, 2024

• flairNLPGH-3429: Fix checkpointing

  jeffpicard

  committed
  952e9fa

  Copy full SHA for 952e9fa

• Merge branch 'master' into multi-gpu

  HallerPatrick

  authored
  fb4f07a

  Copy full SHA for fb4f07a