Skip to content

Commit 5937565

Browse files
committed
Fix import statement for IJEPA class
1 parent 5e19f87 commit 5937565

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmlearn/tasks/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Modules for pretraining, downstream and evaluation tasks."""
22

33
from mmlearn.tasks.contrastive_pretraining import ContrastivePretraining
4-
from mmlearn.tasks.ijepa_pretraining import IJEPA
4+
from mmlearn.tasks.ijepa import IJEPA
55
from mmlearn.tasks.zero_shot_classification import ZeroShotClassification
66
from mmlearn.tasks.zero_shot_retrieval import ZeroShotCrossModalRetrieval
77

0 commit comments

Comments
 (0)