Skip to content

Commit a7f5849

Browse files
authored
Remove CTC loss (moved to NNlib) (#2024)
* remove CTC loss, moved to NNlib * trigger * don't export ctc_loss
1 parent b8bdc2d commit a7f5849

File tree

3 files changed

+4
-376
lines changed

3 files changed

+4
-376
lines changed

Project.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
name = "Flux"
22
uuid = "587475ba-b771-5e3f-ad9e-33799f191a9c"
3-
version = "0.13.4"
3+
version = "0.13.5"
44

55
[deps]
66
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -32,8 +32,8 @@ ChainRulesCore = "1.12"
3232
Functors = "0.2.8"
3333
MLUtils = "0.2"
3434
MacroTools = "0.5"
35-
NNlib = "0.8.2"
36-
NNlibCUDA = "0.2"
35+
NNlib = "0.8.9"
36+
NNlibCUDA = "0.2.4"
3737
Optimisers = "0.2.1"
3838
ProgressLogging = "0.1"
3939
Reexport = "0.2, 1.0"

src/losses/Losses.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ using Zygote: @adjoint
66
using ChainRulesCore
77
using ..Flux: ofeltype, epseltype
88
using CUDA
9-
using NNlib: logsoftmax, logσ
9+
using NNlib: logsoftmax, logσ, ctc_loss, ctc_alpha, ∇ctc_loss
1010
import Base.Broadcast: broadcasted
1111

1212
export mse, mae, msle,
@@ -19,11 +19,9 @@ export mse, mae, msle,
1919
dice_coeff_loss,
2020
poisson_loss,
2121
hinge_loss, squared_hinge_loss,
22-
ctc_loss,
2322
binary_focal_loss, focal_loss, siamese_contrastive_loss
2423

2524
include("utils.jl")
2625
include("functions.jl")
27-
include("ctc.jl")
2826

2927
end #module

0 commit comments

Comments
 (0)