Can you post the errors that you are seeing? Looking at your original post again, it looks like you may be comparing images of different sizes in the loss function.
Hard to tell without seeing the exceptions, though.