mae output normal size images

[yzb1997]

Hello. First of all, I would like to pay tribute to your outstanding work My question is in the MAE code your output is loss and the output image size is much larger than the input image. How can I set the MAE code to output the reconstructed image with the same size as the input image

[lucidrains]

@yzb1997 hey, this may be helpful https://github.com/lucidrains/vit-pytorch/issues/169