MarcoForte / FBA_Matting

Official repository for the paper F, B, Alpha Matting
MIT License
464 stars 95 forks source link

How to export TorchScript file #51

Open shokohigol opened 1 year ago

shokohigol commented 1 year ago

I want to load model in c++ with libtorch, but the libtorch just log jit.save files.

sm = torch.jit.script(model)

not work. Can't convert Union[int,Tuple[int,int]] for MaxUnpool2d.

How to fix this issue.