diff --git a/utilities/utils.py b/utilities/utils.py index f8acdf9..a323bd5 100644 --- a/utilities/utils.py +++ b/utilities/utils.py @@ -8,7 +8,7 @@ def readConfig(file: str) -> dict: with open(file, "r") as f: - return yaml.load(f, Loader=yaml.FullLoader) + return yaml.load(f, Loader=yaml.SafeLoader) def set_seed(seed): torch.backends.cudnn.deterministic = True