From 27e35819fb4e154473638ca44a786b391edb7513 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Fri, 1 Feb 2019 11:25:56 +0800
Subject: [PATCH] fix config of AttentionCluster/LSTM

---
 fluid/PaddleCV/video/configs/attention_cluster.txt | 8 ++++----
 fluid/PaddleCV/video/configs/attention_lstm.txt    | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/fluid/PaddleCV/video/configs/attention_cluster.txt b/fluid/PaddleCV/video/configs/attention_cluster.txt
index f37e4d12..0ce7c4b2 100755
--- a/fluid/PaddleCV/video/configs/attention_cluster.txt
+++ b/fluid/PaddleCV/video/configs/attention_cluster.txt
@@ -15,17 +15,17 @@ topk = 20
 epoch = 5
 learning_rate = 0.001
 pretrain_base = None
-batch_size = 160
+batch_size = 2048
 use_gpu = True
-num_gpus = 4
+num_gpus = 8
 filelist = "dataset/youtube8m/train.list"
 
 [VALID]
-batch_size = 160
+batch_size = 2048
 filelist = "dataset/youtube8m/val.list"
 
 [TEST]
-batch_size = 40
+batch_size = 256
 filelist = "dataset/youtube8m/test.list"
 
 [INFER]
diff --git a/fluid/PaddleCV/video/configs/attention_lstm.txt b/fluid/PaddleCV/video/configs/attention_lstm.txt
index f51e1c01..9154fe2c 100755
--- a/fluid/PaddleCV/video/configs/attention_lstm.txt
+++ b/fluid/PaddleCV/video/configs/attention_lstm.txt
@@ -19,17 +19,17 @@ decay_gamma = 0.1
 weight_decay = 0.0008
 num_samples = 5000000
 pretrain_base = None
-batch_size = 160
+batch_size = 1024
 use_gpu = True
-num_gpus = 4
+num_gpus = 8
 filelist = "dataset/youtube8m/train.list"
 
 [VALID]
-batch_size = 160
+batch_size = 1024
 filelist = "dataset/youtube8m/val.list"
 
 [TEST]
-batch_size = 40
+batch_size = 128
 filelist = "dataset/youtube8m/test.list"
 
 [INFER]
-- 
GitLab