revert: bad rebase

LuxDL · Dec 31, 2024 · c30435f · c30435f
1 parent 8f5c491
commit c30435f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -110,7 +110,7 @@ NNlib = "0.9.26"
 Optimisers = "0.4.1"
 Preferences = "1.4.3"
 Random = "1.10"
-Reactant = "0.2.11"
+Reactant = "0.2.12"
 Reexport = "1.2.2"
 ReverseDiff = "1.15"
 SIMDTypes = "0.1"

diff --git a/docs/Project.toml b/docs/Project.toml
@@ -56,7 +56,7 @@ Optimisers = "0.4.1"
 Pkg = "1.10"
 Printf = "1.10"
 Random = "1.10"
-Reactant = "0.2.11"
+Reactant = "0.2.12"
 StableRNGs = "1"
 StaticArrays = "1"
 WeightInitializers = "1"

diff --git a/lib/LuxLib/Project.toml b/lib/LuxLib/Project.toml
@@ -88,7 +88,7 @@ Octavian = "0.3.28"
 Polyester = "0.7.15"
 Preferences = "1.4.3"
 Random = "1.10"
-Reactant = "0.2.11"
+Reactant = "0.2.12"
 Reexport = "1"
 ReverseDiff = "1.15"
 SLEEFPirates = "0.6.43"

diff --git a/src/helpers/training.jl b/src/helpers/training.jl
@@ -11,7 +11,7 @@ using Static: StaticBool, Static, False, True
 
 using ..Lux: Lux, Utils, ReactantCompatibleOptimisers
 using LuxCore: LuxCore, AbstractLuxLayer
-using MLDataDevices: MLDataDevices, ReactantDevice, get_device_type, cpu_device
+using MLDataDevices: MLDataDevices, ReactantDevice, get_device_type
 
 """
     TrainState