Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
num_layers problem
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