Reproducible Error

main.py

from contextlib import asynccontextmanager

import open3d as o3d
import uvicorn
from fastapi import FastAPI
import torch
import torchvision.models as models

feature_extractor = {}


@asynccontextmanager
async def lifespan(app: FastAPI):
    feature_extractor['fe'] = FeatureExtractor()
    yield


app = FastAPI(lifespan=lifespan)


class FeatureExtractor(torch.nn.Module):
    def __init__(self):
        super(FeatureExtractor, self).__init__()
        self.model = models.vgg16(weights=models.VGG16_Weights.DEFAULT).features
        for param in self.model.parameters():
            param.requires_grad = False

    def forward(self, x):
        x = self.model(x)
        return x


@app.get("/")
async def read_root():
    return {"version": o3d.__version__}


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)