facebookresearch / ijepa

Official codebase for I-JEPA, the Image-based Joint-Embedding Predictive Architecture. First outlined in the CVPR paper, "Self-supervised learning from images with a joint-embedding predictive architecture."
Other
2.83k stars 358 forks source link

Valid mask not found when train on mini image1k dataset #68

Closed oggyfaker closed 4 months ago

oggyfaker commented 4 months ago

Hi authors, i got some bug with this logging 'Valid mask not found, decreasing acceptable-regions' when i use the small part of image1k full. And when i start train command , the code is stuck with self._sample_block_mask in line 144 (multiblock.py)

My structure directory like this. And my config model is below. How can i fix this issue , please let know. Thanks very much !!!

Screenshot 2024-07-28 at 22 08 16 Screenshot 2024-07-28 at 22 08 33 Screenshot 2024-07-28 at 22 08 49

` data: batch_size: 128 color_jitter_strength: 0.0 crop_scale:

oggyfaker commented 4 months ago

sorry my fault , change crop_size to 224 can fix the problem.