Cannot use DETR without downloading backbone

I have fintuned a DETRForObjectDetection model, but when I want to save it using save_pretrained() or push on MLFlow using mlflow.transformers.log_model(), it saves the config and the weights of the model and the processor, but not the backbone’s (resnet101 ) weights.

The problem that I have is that it always need to download the backbone or check its cache to use it. Is there a way to save with the model and its components the backbone, so that it never downloads it from a website and can work fully locally without caching. Thank you in advance ! :slight_smile:

1 Like