Initial removal of cupy as dependency. Replaced with torch. (#86)

* Initial removal of cupy as dependency. Replaced with torch. Fixing wideband notebook. * This runs. * Fixed issue with num_symbols.
TorchDSP · May 3, 2023 · a6e43df · a6e43df
1 parent 3acac5e
commit a6e43df
Show file tree

Hide file tree

Showing 12 changed files with 891 additions and 261 deletions.
diff --git a/examples/03_example_widebandsig53_dataset.ipynb b/examples/03_example_widebandsig53_dataset.ipynb
diff --git a/examples/05_example_wideband_detector.ipynb b/examples/05_example_wideband_detector.ipynb
@@ -287,7 +287,7 @@
     "    max_epochs=epochs,\n",
     "    callbacks=checkpoint_callback,\n",
     "    accelerator=\"gpu\",\n",
-    "    devices=[2],\n",
+    "    devices=[0],\n",
     "    accumulate_grad_batches=4,\n",
     ")\n",
     "trainer.fit(example_model)\n"
@@ -566,7 +566,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.0"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,