mask token of decoder

lightly-ai · Feb 6, 2024 · e712bb5 · e712bb5
1 parent d10d9e9
commit e712bb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/imagenet/vitb16/mae.py b/benchmarks/imagenet/vitb16/mae.py
@@ -123,7 +123,7 @@ def configure_optimizers(self):
         params, params_no_weight_decay = utils.get_weight_decay_parameters(
             [self.backbone, self.decoder]
         )
-        params.append(self.backbone.mask_token)
+        params.append(self.decoder.mask_token)
         optimizer = AdamW(
             [
                 {"name": "mae", "params": params},