Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

num_layers problem

Open rafikg opened this issue 3 years ago • 0 comments

Hi @aladdinpersson having num_layers =2 > 1, break this line and generate RuntimeError: Expected hidden[0] size (2, 1, 1024), got [1, 1, 1024]

https://github.com/aladdinpersson/Machine-Learning-Collection/blob/ac5dcd03a40a08a8af7e1a67ade37f28cf88db43/ML/Pytorch/more_advanced/Seq2Seq_attention/utils.py#L37

rafikg avatar Apr 19 '22 19:04 rafikg