vietnh1009 / Yolo-v2-pytorch

YOLO for object detection tasks
MIT License
373 stars 107 forks source link

Deprecated manual_seed #28

Open Sukanyasingh3 opened 11 months ago

Sukanyasingh3 commented 11 months ago

In the train_coco.py code, the line torch.cuda.manual_seed(123) uses manual_seed, which is deprecated. Instead, you can use manual_seed_all.

   torch.cuda.manual_seed_all(123) if torch.cuda.is_available() else torch.manual_seed(123)