Code

Fix GRU to match pytorch (#2701).#2704

Code

Open

nwhitehead wants to merge 2 commits intotracel-ai:maintracel-ai/burn:mainfrom nwhitehead:dev-fixgrunwhitehead/burn:dev-fixgruCopy head branch name to clipboard

+81-41

Commits

Commits on Jan 15, 2025

• Fix GRU to match pytorch (tracel-ai#2701).

  Show description for 3134f82
  nwhitehead

  committed
  3134f82

  Copy full SHA for 3134f82

Commits on Jan 16, 2025

• Add GruConfig option reset_after to allow both reset behaviors.

  nwhitehead

  committed
  95a7eaf

  Copy full SHA for 95a7eaf