Code

Experimental support for accelerate#826

Code

Merged

BenjaminBossan merged 10 commits intomasterskorch-dev/skorch:masterfrom support-huggingface-accelerateskorch-dev/skorch:support-huggingface-accelerateCopy head branch name to clipboardMar 4, 2022

+403-1

Commits

Commits on Dec 23, 2021

• [WIP] Experimental support for accelerate

  Show description for e070abe
  BenjaminBossan

  committed
  e070abe

Commits on Dec 30, 2021

• Merge branch 'master' into support-huggingface-accelerate

  BenjaminBossan

  committed
  08f495d

Commits on Feb 13, 2022

• More extensive tests, documentation

  BenjaminBossan

  committed
  b541ea0

• Add more documentation about the new feature

  Show description for 9c22965
  BenjaminBossan

  committed
  9c22965

Commits on Feb 18, 2022

• Automatically use accelerator.print by default

  Show description for 9d3e5e0
  BenjaminBossan

  committed
  9d3e5e0

Commits on Feb 21, 2022

• Add nan check to test

  BenjaminBossan

  committed
  f195e8c

Commits on Feb 27, 2022

• Set device default to None with accelerate

  BenjaminBossan

  committed
  58310c1

Commits on Mar 2, 2022

• Fix some issues

  Show description for 8409c05
  BenjaminBossan

  committed
  8409c05

• Fix a bug caused by using closure with GradScaler

  Show description for c4b79b9
  BenjaminBossan

  committed
  c4b79b9

• Add accelerate to dev requirements

  Show description for 08681f8
  BenjaminBossan

  committed
  08681f8