Can I create *. model file from checkpoint file?

[turbosonics]

Hi,

I have some training jobs which are over by wall-time limit before reaching target epoch. In this case, because the training job is killed, output folder didn't create *_compiled/model file or *.model file.

But I hope to test at least some of these training result from LAMMPS MD simulation before I restart this training work.

So I'm curious, how can I generate these model files from checkpoint pt file? Once I have model file then I can convert it to lammps-ready pt file using create_lammps_model.py.

Or are there any other ways to generate lammps-ready model file from training jobs that is finished before reaching target epoch?

Thanks

[ilyes319]

There is no direct wait to do that yet, but the easiest is the following:

• First run a short job, to create a temple *.model that has the same hypers as the final mode.
• Then use a following script:

ckpt = torch.load(path_ckpt)
model = torch.load(path_model)
model.load_state_dict(ckpt["model"])
torch.save(model, path_new_model)