Skip to content

Commit 1184f04

Browse files
authored
update RRR tutorials
1 parent f10b3b1 commit 1184f04

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

notebooks/reduced-rank-model-example.ipynb

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,7 @@
3535
"from lightning.pytorch import Trainer\n",
3636
"from lightning.pytorch.callbacks import ModelCheckpoint\n",
3737
"\n",
38-
"from models.decoders import (\n",
39-
" MultiRegionReducedRankDecoder,\n",
40-
" MultiSessionReducedRankDecoder,\n",
41-
" ReducedRankDecoder,\n",
42-
")\n",
38+
"from models.decoders import ReducedRankDecoder\n",
4339
"\n",
4440
"from utils.config_utils import config_from_kwargs, update_config\n",
4541
"from utils.data_loader_utils import SingleSessionDataModule\n",

0 commit comments

Comments
 (0)