Update README.md

README.md

@@ -56,6 +56,7 @@ final_layer_keys = {
     "final_layer.linear.bias",
 }
 
+
 # a) Contextual downstream architecture
 #    ----------------------------------
 model = SignalJEPA_Contextual(
@@ -74,6 +75,7 @@ FILTERED_model_state_dict = {
     k: v for k, v in model_state_dict.items() if not any(k.startswith(pre) for pre in ["transformer.", "pos_encoder."])
 }
 
+
 # b) Post-local downstream architecture
 #    ----------------------------------
 model = SignalJEPA_PostLocal(
@@ -86,6 +88,7 @@ missing_keys, unexpected_keys = model.load_state_dict(FILTERED_model_state_dict,
 assert unexpected_keys == []
 assert set(missing_keys) == final_layer_keys
 
+
 # c) Pre-local architecture
 #    ----------------------
 model = SignalJEPA_PreLocal(