torchTS icon indicating copy to clipboard operation
torchTS copied to clipboard

Published 20 hours ago • verified publisher icon Rose-STL-Lab

Generalize seq2seq

Open klane opened this issue 4 years ago • 0 comments
trafficstars

Generalize seq2seq class to permit additional recurrent layer types (e.g. GRU). The current implementation has LSTMs hardcoded in both the encoder and decoder.

klane avatar Jun 16 '21 05:06 klane