From f33d27e5e4536acf1901db601be177ea7bbdc309 Mon Sep 17 00:00:00 2001
From: xjqbest <173596896@qq.com>
Date: Mon, 1 Jun 2020 23:57:20 +0800
Subject: [PATCH] fix

---
 models/contentunderstanding/classification/model.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/models/contentunderstanding/classification/model.py b/models/contentunderstanding/classification/model.py
index 45c50b9b..ce9caf5b 100644
--- a/models/contentunderstanding/classification/model.py
+++ b/models/contentunderstanding/classification/model.py
@@ -27,7 +27,8 @@ class Model(ModelBase):
         self.emb_dim = 8
         self.hid_dim = 128
         self.class_dim = 2
-        self.is_sparse = envs.get_global_env("hyper_parameters.is_sparse", False)
+        self.is_sparse = envs.get_global_env("hyper_parameters.is_sparse",
+                                             False)
 
     def input_data(self, is_infer=False, **kwargs):
         data = fluid.data(
@@ -43,7 +44,10 @@ class Model(ModelBase):
         seq_len = input[2]
 
         # embedding layer
-        emb = fluid.embedding(input=data, size=[self.dict_dim, self.emb_dim], is_sparse=self.is_sparse)
+        emb = fluid.embedding(
+            input=data,
+            size=[self.dict_dim, self.emb_dim],
+            is_sparse=self.is_sparse)
         emb = fluid.layers.sequence_unpad(emb, length=seq_len)
         # convolution layer
         conv = fluid.nets.sequence_conv_pool(
-- 
GitLab