Error in initial yaml loading (TypeError: load() missing 1 required positional argument: 'Loader')

[KayembaLuwaga]

In lines [53-59], trainval.py file;
I solved by commenting line 57 in trainval.py file, default_arg = yaml.load(f), and replace with default_arg = yaml.safe_load(f) , as shown below:

def load_arg(p):
# save arg
if os.path.exists(p.config):
with open(p.config, 'r') as f:
# default_arg = yaml.load(f)
default_arg = yaml.safe_load(f)

[zhou00NJUST]

try

pip install pyyaml==5.1