File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 77 "metadata": {},
88 "outputs": [],
99 "source": [
10- "import os\n",
11- "import subprocess\n",
12- "import pytest\n",
13- "from mudata import MuData\n",
10+ "import tempfile\n",
11+ "\n",
1412 "import scanpy as sc\n",
1513 "import scvi\n",
16- "from scvi.model import SCVI\n",
17- "import tempfile\n",
1814 "import seaborn as sns\n",
19- "import torch"
15+ "import torch\n",
16+ "from scvi.model import SCVI"
2017 ]
2118 },
2219 {
270267 "source": [
271268 "print(\"multi GPU SCVI train\")\n",
272269 "model.train(\n",
273- " max_epochs=100,\n",
274- " check_val_every_n_epoch=1,\n",
275- " accelerator=\"gpu\",\n",
276- " devices=-1,\n",
277- " datasplitter_kwargs=datasplitter_kwargs,\n",
278- " strategy=\"ddp_notebook_find_unused_parameters_true\",\n",
279- " )\n",
270+ " max_epochs=100,\n",
271+ " check_val_every_n_epoch=1,\n",
272+ " accelerator=\"gpu\",\n",
273+ " devices=-1,\n",
274+ " datasplitter_kwargs=datasplitter_kwargs,\n",
275+ " strategy=\"ddp_notebook_find_unused_parameters_true\",\n",
276+ ")\n",
280277 "print(\"done\")"
281278 ]
282279 },
490487 " color=[\"donor\", \"cell_source\"],\n",
491488 " ncols=2,\n",
492489 " frameon=False,\n",
493- ")\n "
490+ ")"
494491 ]
495492 },
496493 {
You can’t perform that action at this time.
0 commit comments