Abhiram V commited on
Commit
4501169
·
unverified ·
1 Parent(s): 91c30e4

Update yolo.py with yaml.SafeLoader (#1970)

Browse files
Files changed (1) hide show
  1. models/yolo.py +1 -1
models/yolo.py CHANGED
@@ -71,7 +71,7 @@ class Model(nn.Module):
71
  import yaml # for torch hub
72
  self.yaml_file = Path(cfg).name
73
  with open(cfg) as f:
74
- self.yaml = yaml.load(f, Loader=yaml.FullLoader) # model dict
75
 
76
  # Define model
77
  ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels
 
71
  import yaml # for torch hub
72
  self.yaml_file = Path(cfg).name
73
  with open(cfg) as f:
74
+ self.yaml = yaml.load(f, Loader=yaml.SafeLoader) # model dict
75
 
76
  # Define model
77
  ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels