search

TorchEnsemble-Community / Ensemble-Pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
https://ensemble-pytorch.readthedocs.io
BSD 3-Clause "New" or "Revised" License
1.08k stars 95 forks source link

How to ues the result "ckpt.pth" #149

Open wyxAshdila opened 1 year ago

wyxAshdila commented 1 year ago

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

xuyxu commented 1 year ago

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 commented 3 months ago

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

xuyxu commented 3 months ago

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.