Mismatch between target and output tensor

mckim2020 / StockER

A machine that can analyze and predict stock markets.

2 stars 0 forks source link

Mismatch between target and output tensor #1

Open mckim2020 opened 4 years ago

mckim2020 commented 4 years ago

Size of target tensor is [1, 1, 8] and size of output tensor of the net, which is also an input tensor of MSELoss() is [1, 8] Since this may cause a critical error in the loss calculation process, resetting target tensor's size from [1, 1, 8] to [1, 8] is needed!

ASAP

mckim2020 commented 4 years ago

@KYHSGeekCode

tensor.unsqueeze(1) 을 disable 하여 해결함