wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 18:06:19 +08:00
Code Issues Packages Projects Releases Wiki Activity
446 Commits 1 Branch 0 Tags
master
Commit Graph

2 Commits

Author SHA1 Message Date
wassname 3fb53f620e wip 2022-12-23 14:34:54 +08: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