Update demo.py

AliaksandrSiarohin · Aug 24, 2022 · 5f3dee0 · 5f3dee0
1 parent ca49071
commit 5f3dee0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -24,7 +24,7 @@
 def load_checkpoints(config_path, checkpoint_path, cpu=False):
 
     with open(config_path) as f:
-        config = yaml.load(f)
+        config = yaml.full_load(f)
 
     generator = OcclusionAwareGenerator(**config['model_params']['generator_params'],
                                         **config['model_params']['common_params'])