Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Alexis Brenon
arcades
Commits
feb6f9aa
Commit
feb6f9aa
authored
Jul 17, 2017
by
Alexis Brenon
Browse files
🚑
Fix learn failing
Make sure that default learn_start is greater than the minibatch size
parent
3bfa02c2
Changes
1
Hide whitespace changes
Inline
Side-by-side
arcades/agent/NeuralQLearner.lua
View file @
feb6f9aa
...
...
@@ -92,10 +92,6 @@ function class:__init(args, dump)
self
:
_init_inference_network
(
args
.
inference
,
dump
.
inference
)
--- Number of steps after which learning starts.
-- Add delay to populate the experience pool
-- @tfield number self.learn_start
self
.
learn_start
=
args
.
learn_start
or
(
0
.
1
*
self
.
experience_pool
.
pool
.
max_size
)
--- Learning frequency (epoch size).
-- @tfield number self.update_freq
self
.
update_freq
=
args
.
update_freq
or
12
...
...
@@ -105,6 +101,10 @@ function class:__init(args, dump)
--- Number of minibatch learning during a learning epoch.
-- @tfield number self.n_replay
self
.
n_replay
=
args
.
n_replay
or
1
--- Number of steps after which learning starts.
-- Add delay to populate the experience pool
-- @tfield number self.learn_start
self
.
learn_start
=
args
.
learn_start
or
math.max
(
self
.
minibatch_size
+
1
,
(
0
.
1
*
self
.
experience_pool
.
pool
.
max_size
))
--- Scale rewards delta.
-- @tfield boolean self.rescale_r
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment