Code

implement ctc loss function#1049

Code

Closed

wcshds wants to merge 20 commits intotracel-ai:maintracel-ai/burn:mainfrom wcshds:ctc-losswcshds/burn:ctc-lossCopy head branch name to clipboard

+503

Commits

Commits on Dec 4, 2023

• implement ctc loss function

  wcshds

  committed
  3365b50

• remove vec! macro

  wcshds

  committed
  f92b945

Commits on Dec 5, 2023

• fix wrong indice when assign initial val to alpha

  wcshds

  committed
  4ddfbb3

• update test case

  wcshds

  committed
  92d694d

Commits on Dec 6, 2023

• remove batch loop

  wcshds

  committed
  71fc4df

• cache current_target_primes

  wcshds

  committed
  0813605

• make sure tensors are on the same device

  wcshds

  committed
  dd9cf8b

• use clamp_min instead of mask_where

  wcshds

  committed
  94d52d2

Commits on Dec 8, 2023

• remove squence loop

  wcshds

  committed
  7121852

• use into_scalar in assertions

  wcshds

  committed
  eb5357f

Commits on Dec 11, 2023

• reduce the size of log alphas

  wcshds

  committed
  5c57283

• wordaround for slice bug on libtorch backend tracel-ai#1055

  wcshds

  committed
  7d047cd

Commits on Dec 12, 2023

• reduce the size of one_hot

  wcshds

  committed
  9599f32

• make sure tensors are on the same device

  wcshds

  committed
  25972ce

Commits on Dec 24, 2023

• Merge branch 'tracel-ai:main' into ctc-loss

  wcshds

  authored
  08f837f

• adapt to burn's new device api

  wcshds

  committed
  2758b1b

Commits on Dec 27, 2023

• make sure the argument of the logarithm greater than 0

  wcshds

  committed
  9315d98

• refactor the small value used to prevent log(0) into a constant

  wcshds

  committed
  08dd4eb

• fix typo

  wcshds

  committed
  4f009fc

• remove unnecessary code

  wcshds

  committed
  0b99922