Lama Is a great inpaint model, open source code
At present, it’s bestressult is in CKPT format, and I hope to convert it to onnx mode. I tried the code,
model = load_checkpoint(train_config, checkpoint_path, strict=False, map_location='cpu')
model.freeze()
model.to(device)
torch.save(model,'bestresult.pth')
img = torch.rand(1,3,320,320,requires_grad=False)
img = img.to(torch.device('cpu'))
torch.onnx.export(model,img,'bestresult.onnx',opset_version=11)
print('=====================best onnx result is saved!================')
but it reported an error. How can I solve it? thank you.
[2021-11-26 06:25:35,036][__main__][CRITICAL] - Prediction failed due to too many indices for tensor of dimension 4:
Traceback (most recent call last):
File "bin/predict.py", line 68, in main