From d2b7bac93c5704826a8ca398e8d73ce602acd26d Mon Sep 17 00:00:00 2001 From: gmcather Date: Wed, 11 Apr 2018 12:39:47 +0000 Subject: [PATCH] add train --- fluid/text_classification/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/text_classification/train.py b/fluid/text_classification/train.py index e2514090..0e295882 100644 --- a/fluid/text_classification/train.py +++ b/fluid/text_classification/train.py @@ -85,9 +85,9 @@ def train_net(): batch_size = 128, buf_size = 50000) if sys.argv[1] == "bow": - train(train_reader, word_dict, bow_net, use_cuda=False, + train(train_reader, word_dict, bow_net, use_cuda=True, parallel=False, save_dirname="bow_model", lr=0.002, - pass_num=1, batch_size=128) + pass_num=30, batch_size=128) elif sys.argv[1] == "cnn": train(train_reader, word_dict, cnn_net, use_cuda=True, parallel=False, save_dirname="cnn_model", lr=0.01, -- GitLab