IJEPA

assets/benchmarks/cifar10/byol.yaml

@@ -13,14 +13,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: True
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 128]

assets/benchmarks/cifar10/dino.yaml

@@ -9,14 +9,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: True
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 2048]

assets/benchmarks/cifar100/byol.yaml

@@ -13,14 +13,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: True
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 128]

assets/benchmarks/cifar100/dino.yaml

@@ -9,14 +9,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: True
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 2048]

assets/benchmarks/imagenette/byol.yaml

@@ -13,14 +13,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: False
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 128]

assets/benchmarks/imagenette/dino.yaml

@@ -9,14 +9,14 @@ trainer:
   # ===== Module Parameters =====
   module:
     backbone:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.load_backbone
         name: resnet50
         low_resolution: False
         num_classes: null
     projector:
-      _target_: stable_ssl.modules.TeacherStudentModule
+      _target_: stable_ssl.backbone.utils.TeacherStudentWrapper
       student:
         _target_: stable_ssl.modules.MLP
         sizes: [2048, 2048, 2048]

benchmarks/cifar10/vicreg-resnet18.py

@@ -153,10 +153,11 @@ def forward(self, batch, stage):
 )
 
 wandb_logger = WandbLogger(
-    entity="stable-ssl",
-    project="cifar10-vicreg",
-    name="vicreg-resnet18",
-    log_model=False,
+    project="ijepa-cifar10",
+    entity="samibg",  # Your W&B entity
+    name="vicreg-cifar10-run",
+    log_model=False,  # Set to True if you want to save model artifacts
+    offline=False,  # Ensure offline mode
 )
 
 trainer = pl.Trainer(
@@ -165,6 +166,7 @@ def forward(self, batch, stage):
     callbacks=[knn_probe, linear_probe],
     precision="16-mixed",
     logger=wandb_logger,
+    devices=1,
     enable_checkpointing=False,
 )