search

median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
1.94k stars 181 forks source link

Saving and loading models #48

Closed yamizi closed 1 year ago

yamizi commented 1 year ago

Hello LibMTL team, I went through your code, and I am not sure which part could be extended to support loading and saving the models to disk. The model inside the Trainer is not picklable, and I was wondering if you could hint me to the components that would need to be saved and loaded for instance to resume training or postpone testing, Best, Salah

Baijiong-Lin commented 1 year ago

Please refer to https://github.com/median-research-group/LibMTL/issues/23.

Baijiong-Lin commented 1 year ago

v1.1.5 now supports saving and loading models.

Baijiong-Lin commented 1 year ago

Closed as no further updates.