19reborn / NeuS2

[ICCV 2023] Official code for NeuS2
Other
601 stars 38 forks source link

Load learned model in Pytorch #33

Open spock-the-wizard opened 10 months ago

spock-the-wizard commented 10 months ago

Hi,

thank you for the great work. I am curious how to use the learned model in Pytorch. Is there a way I could perhaps automatically redefine the network in Pytorch and load the trained weights there?

I have tried opening the checkpoints/xxx.msgpack file and loading it to a pytorch network, but am out of luck. Any help would be very helpful!

2454511550Lin commented 8 months ago

Also curious about this