search

zhou13 / lcnn

LCNN: End-to-End Wireframe Parsing
MIT License
506 stars 94 forks source link

Size of PTH file? #71

Closed I1234567R closed 1 year ago

I1234567R commented 1 year ago

Thank you for this repository it's greatly appreciated! The example pth file provided on Google Drive is 152,753 KB compared with the same data trained locally at 153,159 KB. Has anything changed since that model was trained that would cause this increase? I need to be able to train on my custom dataset and produce exactly the same model file as the example. Currently, I get the larger of the two files even with the example dataset.

Any help/information is greatly appreciated.

zhou13 commented 1 year ago

I think it is might be due to the difference in pytorch version.

I1234567R commented 1 year ago

Do you know what the exact version of pytorch was used? I've followed the instructions however this installs the latest version.

zhou13 commented 1 year ago

Unfortunately, I don't have the record. I generally won't suggest this but If you need the exact file size, you may try the pytorch version release when the repo was created.

I1234567R commented 1 year ago

Thank you for the advice, this was the problem! Also, have you written anything to convert the pth models to pt or onnx so they can be run in another language, ideally C++?

zhou13 commented 1 year ago

I haven't worked on deployment of this project.