Update trainer.py

NVlabs · Jul 9, 2023 · a3bbfb4 · a3bbfb4
1 parent d0e8b43
commit a3bbfb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trainer.py b/trainer.py
@@ -162,7 +162,7 @@ def load_network(network, name):
 def load_config(name):
     config_path = os.path.join('./models',name,'opts.yaml')
     with open(config_path, 'r') as stream:
-        config = yaml.load(stream)
+        config = yaml.safe_load(stream)
     return config