Afternoon,
I have a Autoencoder working now, but at the start i get the following warning:
UserWarning: Using a target size (torch.Size([16, 2048])) that is different to the input size (torch.Size([16, 1, 2048])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
16 is my batch size and the vector length is 2048, so i would expect to see 16,1,2048. I cannot understand where 16,2048 comes from…
how do I reshape the target size to 16,1,2048 to solve the probelm so that the warning goes away?
It looks like the input vector size is 16,2048, and using data=data.unsqueeze(1) changes this to 16,1,2048 but i still have the warning…
thanks,
Chaslie