Parcourir la source
Security Fix for Arbitrary Code Execution - huntr.dev (#1962)
Co-authored-by: Anon-Artist <61599526+Anon-Artist@users.noreply.github.com> Co-authored-by: Jamie Slome <jamie@418sec.com>5.0
huntr.dev | the place to protect open source
GitHub
il y a 3 ans
1 fichiers modifiés avec 1 ajouts et 1 suppressions
+ 1
- 1
train.py
Voir le fichier
| @@ -59,7 +59,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'] | |||
Chargement…