search

median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
2.01k stars 185 forks source link

Time-series MTL #4

Closed madhu-kt closed 2 years ago

madhu-kt commented 2 years ago

Hi,

First of all, this is a fantastic library! Amazing work.

My question: can LibMTL can be used in time-series applications? I.e. do we only need to provide encoder/decoder architectures such as LSTMs? Would the loss-weighting methods need to be extended in any way? I'm fairly new to MTL, so pardon the naive question.

Thanks, Madhu

Baijiong-Lin commented 2 years ago

Yes, you just need to provide appropriate encoder/decoder architectures to LibMTL according to your problem.

madhu-kt commented 2 years ago

Sounds good, thanks!