search

ShunChengWu / 3DSSG

Other
142 stars 22 forks source link

About Trace mode #52

Closed k2room closed 6 months ago

k2room commented 10 months ago

Hello, thank you for the great work!

I tried to train a modified SGFN and run the model on incremental C++ framework.

I want to create files for the trace folder like traced/gcn_1_edgeatten_prop, traced/obj_pnetenc, ... etc.

How can I convert the trained model checkpoint (model_best.pt) to CVPR21_traced files format?

Can you tell me about the format or file type of the CVPR21_traced model?

Is there some code for traced mode?

ShunChengWu commented 7 months ago

To do that you will need to use the old branch https://github.com/ShunChengWu/3DSSG/tree/cvpr21 switch to the old branch and check the readme file. It's a recursive way of generating the traced model. The main model will call trace on the submodules. I will recommend you to read the code yourself which is often easier to understand.