Skip to content

Commit 3f17c87

Browse files
committed
fix params not defined issues
1 parent 8943dd7 commit 3f17c87

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/Algorithm/DDPG.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
## TODO: Need reconstruct !!!
22
using Flux: glorot_normal, glorot_uniform
33
using StableRNGs
4-
using Flux.Losses
54
using IntervalSets
65

6+
Base.copyto!(dest::ReinforcementLearning.NeuralNetworkApproximator, src::ReinforcementLearning.NeuralNetworkApproximator) =
7+
Flux.loadparams!(dest.model, Flux.params(src))
8+
79
function Base.rsplit( v, l::Int)
810
u = reshape(v,l,length(v)÷l)
911
[u[:,i] for i=1:size(u,2)]

0 commit comments

Comments
 (0)