diff --git a/Project.toml b/Project.toml index 1aad05b..12c765b 100644 --- a/Project.toml +++ b/Project.toml @@ -26,11 +26,11 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [compat] BoundaryValueDiffEq = "2" -Convex = "0.14, 0.15" +Convex = "0.15" Distributions = "0.25" -Flux = "0.12, 0.14" -Interpolations = "0.13, 0.15" -IntervalSets = "0.5, 0.7" +Flux = "0.12, 0.13" +Interpolations = "0.13" +IntervalSets = "0.5" QuadGK = "2" ReinforcementLearning = "0.10" SCS = "1" diff --git a/src/Algorithm/DDPG.jl b/src/Algorithm/DDPG.jl index 5f353f2..f926545 100644 --- a/src/Algorithm/DDPG.jl +++ b/src/Algorithm/DDPG.jl @@ -1,9 +1,11 @@ ## TODO: Need reconstruct !!! using Flux: glorot_normal, glorot_uniform using StableRNGs -using Flux.Losses using IntervalSets +Base.copyto!(dest::ReinforcementLearning.NeuralNetworkApproximator, src::ReinforcementLearning.NeuralNetworkApproximator) = + Flux.loadparams!(dest.model, Flux.params(src)) + function Base.rsplit( v, l::Int) u = reshape(v,l,length(v)÷l) [u[:,i] for i=1:size(u,2)]