mlp layer + new grid

2026-05-24 09:46:54 +02:00
parent 68505baacb
commit 825281b65a
5 changed files with 55 additions and 11 deletions
@@ -6,6 +6,9 @@ meta:
  read_checkpoint: jepa-ep300.pth.tar
  use_bfloat16: true
  num_classes: 182
+  probe_type: linear
+  mlp_hidden_dim: 1024
+  dropout: 0.0

 data:
  batch_size: 256