# Run detectron2 for feature extraction in SageMaker notebook

**URL:** <https://discuss.huggingface.co/t/run-detectron2-for-feature-extraction-in-sagemaker-notebook/15569>\
**Category:** Amazon SageMaker\
**Created:** [March 10, 2022, 10:49am UTC](https://discuss.huggingface.co/t/run-detectron2-for-feature-extraction-in-sagemaker-notebook/15569 "2022-03-10T10:49:20Z")\
**Posts on this page:** 1\
**Showing post:** 2

<div class="post-metadata">

**Author:** ![philschmid](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/philschmid/32/38022_2.png) [@philschmid](https://discuss.huggingface.co/u/philschmid)\
**Post date:** [March 10, 2022, 1:46pm UTC](https://discuss.huggingface.co/t/run-detectron2-for-feature-extraction-in-sagemaker-notebook/15569/2 "2022-03-10T13:46:34Z")

</div>

@Petrus you can take a look at [Sagemaker Serverless Inference for LayoutLMv2 model - #15 by mansimov](https://discuss.huggingface.co/t/sagemaker-serverless-inference-for-layoutlmv2-model/14186/15)  
to see how to install detectron2 properly.

Are you trying to fine-tune VisualBERT in a SageMaker Notebook or using the `HuggingFace.fit()` estimator=

---

_[View the full topic](https://discuss.huggingface.co/t/run-detectron2-for-feature-extraction-in-sagemaker-notebook/15569)._