Masked AutoEncoder Reconstruction

tarakzai: <p>I’m working with MAE and I have used the pre-trained MAE to train on my data which are images of roots.I have trained the model on 2000 images for 200 epochs but when I input an image to the model and visualise the reconstruction it’s only a blackish image and nothing else.<br>
Are there any suggestions<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/0/2/02c7003acb3b8334cbe32a8af6684b92a0490a25.png" data-download-href="https://discuss.pytorch.org/uploads/default/02c7003acb3b8334cbe32a8af6684b92a0490a25" title="pytorch start"><img src="https://discuss.pytorch.org/uploads/default/original/3X/0/2/02c7003acb3b8334cbe32a8af6684b92a0490a25.png" alt="pytorch start" data-base62-sha1="ozjmQRNu9kQZ9eVul38AehkbOJ" width="690" height="487" data-dominant-color="F1F1F2"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">pytorch start</span><span class="informations">1051×742 36 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div><br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/0/3/038062f6876b5b6715a5f1c5d63fbe478c326185.png" data-download-href="https://discuss.pytorch.org/uploads/default/038062f6876b5b6715a5f1c5d63fbe478c326185" title="pytorch 2"><img src="https://discuss.pytorch.org/uploads/default/original/3X/0/3/038062f6876b5b6715a5f1c5d63fbe478c326185.png" alt="pytorch 2" data-base62-sha1="uYuXBLP4vp5CaPYqff95mxkDuR" width="582" height="500" data-dominant-color="F2F3F3"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">pytorch 2</span><span class="informations">830×713 18.4 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div><br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/1/2/12d950da25cbcbb373290c8457350ea9a558d04c.png" data-download-href="https://discuss.pytorch.org/uploads/default/12d950da25cbcbb373290c8457350ea9a558d04c" title="pytorch forum 1"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/1/2/12d950da25cbcbb373290c8457350ea9a558d04c_2_613x500.png" alt="pytorch forum 1" data-base62-sha1="2GKbS3Vqhoocz3w5FlutspnhLCk" width="613" height="500" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/1/2/12d950da25cbcbb373290c8457350ea9a558d04c_2_613x500.png, https://discuss.pytorch.org/uploads/default/original/3X/1/2/12d950da25cbcbb373290c8457350ea9a558d04c.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/1/2/12d950da25cbcbb373290c8457350ea9a558d04c.png 2x" data-dominant-color="DDDCDB"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">pytorch forum 1</span><span class="informations">911×742 158 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div><br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/5/a/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4.png" data-download-href="https://discuss.pytorch.org/uploads/default/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4" title="pytorch forum 2"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/5/a/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4_2_690x488.png" alt="pytorch forum 2" data-base62-sha1="cWugrHvec3QRD5lPU9lVt6WLb6c" width="690" height="488" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/5/a/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4_2_690x488.png, https://discuss.pytorch.org/uploads/default/original/3X/5/a/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/5/a/5ab6a278591c083e1cf5089f7fbb80c7fb3122d4.png 2x" data-dominant-color="D2D1D0"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">pytorch forum 2</span><span class="informations">772×546 50 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>