Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

implement ctc loss function #1049

Closed
wants to merge 20 commits into from
Closed

implement ctc loss function #1049

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
3365b50
implement ctc loss function
wcshds Dec 4, 2023
f92b945
remove vec! macro
wcshds Dec 4, 2023
4ddfbb3
fix wrong indice when assign initial val to alpha
wcshds Dec 5, 2023
92d694d
update test case
wcshds Dec 5, 2023
71fc4df
remove batch loop
wcshds Dec 6, 2023
0813605
cache current_target_primes
wcshds Dec 6, 2023
dd9cf8b
make sure tensors are on the same device
wcshds Dec 6, 2023
94d52d2
use clamp_min instead of mask_where
wcshds Dec 6, 2023
7121852
remove squence loop
wcshds Dec 8, 2023
eb5357f
use into_scalar in assertions
wcshds Dec 8, 2023
5c57283
reduce the size of log alphas
wcshds Dec 11, 2023
7d047cd
wordaround for slice bug on libtorch backend #1055
wcshds Dec 11, 2023
9599f32
reduce the size of one_hot
wcshds Dec 12, 2023
25972ce
make sure tensors are on the same device
wcshds Dec 12, 2023
08f837f
Merge branch 'tracel-ai:main' into ctc-loss
wcshds Dec 24, 2023
2758b1b
adapt to burn's new device api
wcshds Dec 24, 2023
9315d98
make sure the argument of the logarithm greater than 0
wcshds Dec 27, 2023
08dd4eb
refactor the small value used to prevent log(0) into a constant
wcshds Dec 27, 2023
4f009fc
fix typo
wcshds Dec 27, 2023
0b99922
remove unnecessary code
wcshds Dec 27, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading