diff --git a/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py b/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
index cbaf222ec..1b741cedd 100644
--- a/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
+++ b/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
@@ -41,7 +41,9 @@
 
 def main():
     rng  = np.random.RandomState(23456)
-    sess = tf.Session()
+    config = tf.ConfigProto()
+    config.gpu_options.allow_growth = True
+    sess = tf.Session(config=config)
     network = MainNN(
                 rng,
                 sess,