Could you post the shapes of your input, target and output tensor?
Currently it looks like e.g. the batch dimension might be missing and thus an internal broadcasting might happen (in the latest version we’ve added a warning to nn.MSELoss in case this happens).
Also note, that you should pass the training argument to F.dropout, since in your current code dropout will be used during training and evaluation.