From 3d210701608761a180e0c1216cc5f7357dd0e0aa Mon Sep 17 00:00:00 2001
From: Anon-Artist <61599526+Anon-Artist@users.noreply.github.com>
Date: Mon, 21 Dec 2020 09:09:29 +0000
Subject: [PATCH] Update train.py

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 8234025212b8..d22951aee4d1 100644
--- a/train.py
+++ b/train.py
@@ -66,7 +66,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
     cuda = device.type != 'cpu'
     init_seeds(2 + rank)
     with open(opt.data) as f:
-        data_dict = yaml.load(f, Loader=yaml.FullLoader)  # data dict
+        data_dict = yaml.load(f, Loader=yaml.SafeLoader)  # data dict
     with torch_distributed_zero_first(rank):
         check_dataset(data_dict)  # check
     train_path = data_dict['train']