Update pretrain_IJEPA.py

gaasher / I-JEPA

Implementation of I-JEPA from "Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture"

MIT License

249 stars 24 forks source link

Update pretrain_IJEPA.py #1

Closed AymenSe closed 1 year ago

AymenSe commented 1 year ago

Add the m_start_end parms as an attribute for the class IJEPA, because it's producing an error while training: AttributeError: 'IJEPA' object has no attribute 'm_start_end