fix: remove old LossFunctions.jl dispatches

LuxDL · Oct 9, 2024 · d958283 · d958283
1 parent 7c842fd
commit d958283
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 28 deletions.
diff --git a/ext/LuxReactantExt/LuxReactantExt.jl b/ext/LuxReactantExt/LuxReactantExt.jl
@@ -1,17 +1,14 @@
 module LuxReactantExt
 
 using Enzyme: Enzyme, Const, Duplicated, Active
-using LossFunctions: LossFunctions
 using Optimisers: Optimisers
 using Reactant: Reactant, @compile, TracedRArray
 using Setfield: @set!
 using Static: False
-using Statistics: mean
 
-using Lux: Lux, LuxOps, Training, LossFunctionImpl
+using Lux: Lux, LuxOps, Training
 using Lux.Training: TrainingBackendCache, ReactantBackend
 
-include("overrides.jl")
 include("training.jl")
 
 end
diff --git a/ext/LuxReactantExt/overrides.jl b/ext/LuxReactantExt/overrides.jl