uvipen/Yolo-v2-pytorch

deprecated Variable class

Opened this issue · 0 comments

The test_coco_video.py code uses the Variable class, which is deprecated in recent versions of PyTorch. Instead, torch.tensor can be used directly.

  data = torch.tensor(image, dtype=torch.float32)