gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01: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
2018-11-08 16:53:47 +02:00
committed by GitHub
parent 389c65cbbe
commit 3a0a1159e9
11 changed files with 33 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/architectures/mxnet_components/middlewares/middleware.py
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: