How to ues the result "ckpt.pth"

[wyxAshdila]

I use the examples obtaining the modle parameters "ckpt.pth", how can i use it to test my datasets? thanks for your help.

[xuyxu]

Hi @wyxAshdila, sorry for the late response. You can use from torchensemble.utils import io to load the model. Please refer to the example here.

[johnny12150]

Hi @xuyxu, can I fine-tune the model after loading it?

[xuyxu]

Hi @xuyxu, can I fine-tune the model after loading it?

@johnny12150 Hi, sorry for the late response. I am not sure if I can understand the exact meaning of 'fine-tune' from you. If you could encapsulate the dataset to fine-tune into a train loader, I think you could directly train it further after loading a model checkpoint already trained.