提交 5e416847 编写于 作者: B baiyfbupt

code fix

上级 f3bb87d7
......@@ -300,6 +300,7 @@ class EncoderLayer(Layer):
]
self.gumbel_alphas = to_variable(
np.array(mrpc_arch).astype(np.float32))
self.gumbel_alphas.stop_gradient = True
print("gumbel_alphas: \n", self.gumbel_alphas.numpy())
def forward(self, enc_input_0, enc_input_1, epoch, flops=[],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册