@Ronit-j Besides trying out the post that Martin linked, could you also share the error you see, and the code to reproduce the error?
We are working on adding load_state_dict to C++ frontend, but one caveat is that the exact same model structure needs to also be defined in C++ frontend (i.e. you’ll need to write C++ code that defines the same model structure as the Python model). If this is undesirable, you can also trace the model using JIT and then load it into C++ frontend (https://pytorch.org/tutorials/advanced/cpp_export.html 23 is a good guide on this).