From 670802923157fb75fc3aa2578980e85b50d8b408 Mon Sep 17 00:00:00 2001
From: "Harish.K" <harish2704@gmail.com>
Date: Mon, 22 Jan 2018 18:53:02 +0530
Subject: [PATCH] * Fix Bidirectional layer

---
 src/Model.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Model.js b/src/Model.js
index 9cfb585d..170606d7 100644
--- a/src/Model.js
+++ b/src/Model.js
@@ -382,7 +382,7 @@ export default class Model {
       )
       weightNames = forwardWeightNames.concat(backwardWeightNames)
     } else if (layerClass === 'TimeDistributed') {
-      weightNames = layer.layer.params.map(param => `${layerConfig.name}/${param}`)
+      weightNames = layer.wrappedLayer.params.map(param => `${layerConfig.name}/${param}`)
     } else {
       weightNames = layer.params.map(param => `${layerConfig.name}/${param}`)
     }