diff --git a/model.py b/model.py
index 14cc371..94dfa34 100644
--- a/model.py
+++ b/model.py
@@ -65,7 +65,7 @@ def _variable_with_weight_decay(name, shape, wd):
       Variable Tensor
     """
     var = _variable_on_cpu(name, shape,
-                           initializer=tf.contrib.layers.xavier_initializer())
+                           initializer=tf.contrib.layers.xavier_initializer(uniform=False))
     if wd:
         weight_decay = tf.multiply(tf.nn.l2_loss(var), wd, name='weight_loss')
         tf.add_to_collection('losses', weight_decay)
diff --git a/test.py b/test.py
index 4ed7c36..b030f40 100644
--- a/test.py
+++ b/test.py
@@ -55,7 +55,7 @@ def test(dataset, ckptfile):
         saver = tf.train.Saver(tf.all_variables(), max_to_keep=1000)
 
         init_op = tf.global_variables_initializer()
-        sess = tf.Session(config=tf.ConfigProto(log_device_placement=FLAGS.log_device_placement))
+        sess = tf.Session(config=tf.ConfigProto(log_device_placement=FLAGS.log_device_placement, gpu_options=tf.GPUOptions(allow_growth=True)))
 
         saver.restore(sess, ckptfile)
         print 'restore variables done'
diff --git a/train.py b/train.py
index 5dc602d..9402d60 100644
--- a/train.py
+++ b/train.py
@@ -72,7 +72,7 @@ def train(dataset_train, dataset_val, ckptfile='', caffemodel=''):
         saver = tf.train.Saver(tf.all_variables(), max_to_keep=1000)
 
         init_op = tf.global_variables_initializer()
-        sess = tf.Session(config=tf.ConfigProto(log_device_placement=FLAGS.log_device_placement))
+        sess = tf.Session(config=tf.ConfigProto(log_device_placement=FLAGS.log_device_placement, gpu_options=tf.GPUOptions(allow_growth=True)))
         
         if is_finetune:
             # load checkpoint file