LaminarNet-50m v0.6.2 — best checkpoint (step 115500, val_loss=3.7881)

README.md

@@ -91,10 +91,10 @@ config = LaminarNetConfig(
 )
 model = LaminarNet(config)
 
-# Load weights
+# Load weights (head.weight is tied to tok_emb.weight, so strict=False)
 from safetensors.torch import load_file
 state = load_file(model_path)
-model.load_state_dict(state)
+model.load_state_dict(state, strict=False)
 model.eval()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")