Fix Mixed tensor devices in running README.md

mlverse / tft

R implementation of Temporal Fusion Transformers

https://mlverse.github.io/tft/

Other

26 stars 9 forks source link

Fix Mixed tensor devices in running README.md #12

Closed cregouby closed 3 years ago

cregouby commented 3 years ago

[x] fix tft_fit() : make self$device available for the get_decoder_mask()

[x] fix predict() :

Error in (function (self, mat1, mat2, beta, alpha)  : 
#   Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

fix #9