deploy: braindecode/braindecode@6332ba0

Author: PierreGtch

dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip

@@ -105,7 +105,7 @@
       },
       "outputs": [],
       "source": [
-        "model = SignalJEPA_PreLocal.from_pretrained(\n    \"braindecode/SignalJEPA-PreLocal-pretrained\",\n    n_outputs=len(classes),\n)\nprint(model)"
+        "model = SignalJEPA_PreLocal.from_pretrained(\n    \"braindecode/signal-jepa_without-chans\",\n    n_chans=19,\n    n_times=256,\n    n_outputs=len(classes),\n    strict=False,\n)\nprint(model)"
       ]
     },
     {
@@ -177,7 +177,7 @@
       },
       "outputs": [],
       "source": [
-        "clf = EEGClassifier(\n    \"SignalJEPA_PreLocal\",\n    criterion=torch.nn.CrossEntropyLoss,\n    optimizer=torch.optim.AdamW,\n    optimizer__lr=0.005,\n    batch_size=16,\n    callbacks=[\n        \"accuracy\",\n        WeightsLoader(\n            url=\"https://huggingface.co/braindecode/SignalJEPA/resolve/main/signal-jepa_16s-60_adeuwv4s.pth\"\n        ),\n        Freezer(patterns=\"feature_encoder.*\"),\n    ],\n    classes=range(3),\n)\n_ = clf.fit(windows_dataset, y=metadata[\"target\"], epochs=10)"
+        "clf = EEGClassifier(\n    \"SignalJEPA_PreLocal\",\n    criterion=torch.nn.CrossEntropyLoss,\n    optimizer=torch.optim.AdamW,\n    optimizer__lr=0.005,\n    batch_size=16,\n    callbacks=[\n        \"accuracy\",\n        WeightsLoader(\n            url=\"https://huggingface.co/braindecode/signal-jepa_without-chans/resolve/main/pytorch_model.bin\"\n        ),\n        Freezer(patterns=\"feature_encoder.*\"),\n    ],\n    classes=range(3),\n)\n_ = clf.fit(windows_dataset, y=metadata[\"target\"], epochs=10)"
       ]
     },
     {