Cannot find SpatialGN in docker image

facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

Apache License 2.0

26.27k stars 5.45k forks source link

Cannot find SpatialGN in docker image #418

Open dccho opened 6 years ago

dccho commented 6 years ago

I use Detectron docker image. In this docker image, it seems that caffe2 is not latest version. There is no SpatialGN function. How can i update caffe2 in detectron docker image?

ppwwyyxx commented 6 years ago

All the new features in Caffe2 after it has been moved under pytorch repository, haven't been pushed to the public docker image. For now, to use SpatialGN you'll need to rebuild caffe2.

Also, some instructions on building latest caffe2 docker image can be found here: https://github.com/pytorch/pytorch/issues/7450