Feature Suggestion: Support for Pytorch 1.0

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

MIT License

2.82k stars 565 forks source link

Open yzhq97 opened 5 years ago

yzhq97 commented 5 years ago

Hi, torchvision 0.3 has been released and supports ops such as nms, roialign. It is now possible to support PyTorch 1.0.