Convert Pytorch model into Keras

torch.package / torch::deploy
1

I am trying to convert YOLOV5 Pytorch model into Keras. While loading the pretrained model using torch.load getting the following error: ModuleNotFoundError: No module named ‘models’
In pytorch hub YOLO is not available so converting or loading Yolo model is supported or not?

model_path = '/content/yolov5s.pt' 
# Load the weights 
state_dict = torch.load(model_path)

Thanks!

2

this worked for me

model = torch.load(model_path, map_location="cpu").eval()
model.state_dict()

I am also looking on how to convert a pytorch pt file model into Keras.