diff --git a/models/ModelBase.py b/models/ModelBase.py
index b823a74..25757b3 100644
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -45,7 +45,7 @@ class ModelBase(object):
                 device_args['force_gpu_idx'] = io.input_int("Which GPU idx to choose? ( skip: best GPU ) : ", -1, [ x[0] for x in idxs_names_list] )
         self.device_args = device_args
 
-        self.device_config = nnlib.DeviceConfig(allow_growth=False, **self.device_args)
+        self.device_config = nnlib.DeviceConfig(allow_growth=True, **self.device_args)
 
         io.log_info ("Loading model...")