gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-13 18:22:29 +02:00
Code Issues Projects Releases Wiki Activity

fixing the dropout rate code (#72)

Browse Source 
addresses issue #53
This commit is contained in:
Itai Caspi authored and GitHub committed 2018-11-08 16:53:47 +02:00
1 parent 389c65cbbe
commit 3a0a1159e9
11 files changed
+33 -33
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

rl_coach/architectures/mxnet_components/embedders/embedder.py
+2 -2
View File
@@ -39,8 +39,8 @@ class InputEmbedder(nn.HybridBlock):
self.net.add(nn.BatchNorm())
if params.activation_function:
self.net.add(nn.Activation(params.activation_function))
if params.dropout:
self.net.add(nn.Dropout(rate=params.dropout))
if params.dropout_rate:
self.net.add(nn.Dropout(rate=params.dropout_rate))
@property
def schemes(self) -> dict:
rl_coach/architectures/mxnet_components/middlewares/middleware.py
+2 -2
View File
@@ -36,8 +36,8 @@ class Middleware(nn.HybridBlock):
self.net.add(nn.BatchNorm())
if params.activation_function:
self.net.add(nn.Activation(params.activation_function))
if params.dropout:
self.net.add(nn.Dropout(rate=params.dropout))
if params.dropout_rate:
self.net.add(nn.Dropout(rate=params.dropout_rate))
@property
def schemes(self) -> dict: