Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
monticore
EmbeddedMontiArc
generators
CNNArch2Gluon
Commits
2a05ee34
Commit
2a05ee34
authored
Feb 04, 2021
by
Sascha Dewes
Browse files
only use critic_initializer for ddpg and td3
parent
e9a961af
Pipeline
#403964
failed with stage
in 53 seconds
Changes
4
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
src/main/resources/templates/gluon/reinforcement/Trainer.ftl
View file @
2a05ee34
...
...
@@ -93,6 +93,7 @@ if __name__ == "__main__":
<#
else
>
initializer = mx.init.Normal()
</#
if
>
<#
if
config.rlAlgorithm=="ddpg" || config.rlAlgorithm=="td3">
<#
if
(config.configuration.criticInitializer)??>
critic_initializer_params =
{
<#
list
config
.criticInitializerParams
?
keys
as
param
>
...
...
@@ -103,11 +104,12 @@ if __name__ == "__main__":
<#
else
>
critic_initializer = mx.init.Normal()
</#
if
>
</#
if
>
<#
if
config.rlAlgorithm == "dqn">
qnet_creator = CNNCreator_$
{
config
.instanceName
}
.CNNCreator_$
{
config
.instanceName
}
()
qnet_creator.setWeightInitializer(initializer)
qnet_creator.construct(context)
<#
else
>
<#
else
if
config.rlAlgorithm=="ddpg" || config.rlAlgorithm=="td3"
>
actor_creator = CNNCreator_$
{
config
.instanceName
}
.CNNCreator_$
{
config
.instanceName
}
()
actor_creator.setWeightInitializer(initializer)
actor_creator.construct(context)
...
...
src/test/resources/target_code/ReinforcementConfig1/CNNTrainer_reinforcementConfig1.py
View file @
2a05ee34
...
...
@@ -53,7 +53,6 @@ if __name__ == "__main__":
context
=
mx
.
cpu
()
initializer
=
mx
.
init
.
Normal
()
critic_initializer
=
mx
.
init
.
Normal
()
qnet_creator
=
CNNCreator_reinforcementConfig1
.
CNNCreator_reinforcementConfig1
()
qnet_creator
.
setWeightInitializer
(
initializer
)
qnet_creator
.
construct
(
context
)
...
...
src/test/resources/target_code/ReinforcementConfig2/CNNTrainer_reinforcementConfig2.py
View file @
2a05ee34
...
...
@@ -47,7 +47,6 @@ if __name__ == "__main__":
context
=
mx
.
cpu
()
initializer
=
mx
.
init
.
Normal
()
critic_initializer
=
mx
.
init
.
Normal
()
qnet_creator
=
CNNCreator_reinforcementConfig2
.
CNNCreator_reinforcementConfig2
()
qnet_creator
.
setWeightInitializer
(
initializer
)
qnet_creator
.
construct
(
context
)
...
...
src/test/resources/target_code/ReinforcementConfig3/CNNTrainer_reinforcementConfig3.py
View file @
2a05ee34
...
...
@@ -54,7 +54,6 @@ if __name__ == "__main__":
context
=
mx
.
cpu
()
initializer
=
mx
.
init
.
Normal
()
critic_initializer
=
mx
.
init
.
Normal
()
qnet_creator
=
CNNCreator_reinforcementConfig3
.
CNNCreator_reinforcementConfig3
()
qnet_creator
.
setWeightInitializer
(
initializer
)
qnet_creator
.
construct
(
context
)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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