提交 6bb1a961 编写于 作者: G gmcather

fix code style3

上级 99ad2383
...@@ -46,4 +46,3 @@ if __name__ == "__main__": ...@@ -46,4 +46,3 @@ if __name__ == "__main__":
for i in range(30): for i in range(30):
epoch_path = model_path + "/" + "epoch" + str(i) epoch_path = model_path + "/" + "epoch" + str(i)
infer(test_reader, use_cuda=False, model_path=epoch_path) infer(test_reader, use_cuda=False, model_path=epoch_path)
...@@ -125,5 +125,6 @@ def train_net(): ...@@ -125,5 +125,6 @@ def train_net():
print("network name cannot be found!") print("network name cannot be found!")
sys.exit(1) sys.exit(1)
if __name__ == "__main__": if __name__ == "__main__":
train_net() train_net()
...@@ -91,7 +91,7 @@ def prepare_data(data_type="imdb", ...@@ -91,7 +91,7 @@ def prepare_data(data_type="imdb",
batch_size=batch_size) batch_size=batch_size)
elif data_type == "tiny_imdb": elif data_type == "tiny_imdb":
train_reader=paddle.batch( train_reader = paddle.batch(
paddle.reader.shuffle( paddle.reader.shuffle(
tiny_imdb.train(word_dict), buf_size=buf_size), tiny_imdb.train(word_dict), buf_size=buf_size),
batch_size=batch_size) batch_size=batch_size)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册