wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-09-10 12:37:53 +08:00
Code Issues Packages Projects Releases Wiki Activity
446 Commits 1 Branch 0 Tags
master
Commit Graph
2 Commits
Author SHA1 Message Date
Kashif Rasul cc7dea9f2f Fixed TFT transform 2021-07-07 11:22:20 +02:00
Kashif Rasul 0053eceaf1 Temporal fusion transformer (#40) 
* initial tft model

* fixes

* fix FeatureProjector

* fixed attention module

* added quantile loss output

* added predict

* added tft_transform from gluonts

* requires tensor_split

* comment out key_padding_mask

give nans on GPU

* added example
 2021-04-15 19:10:55 +02:00