commit f6f7761d197e88f491a69a3a77d89ee35f937be5
parent f45973583e167ca206c84853358d3a2053c18b1c
Author: Steven Atkinson <steven@atkinson.mn>
Date: Tue, 13 Dec 2022 22:48:42 -0600
Add TensorBoard monitoring to easy_colab.ipynb (#59)
Diffstat:
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/bin/train/easy_colab.ipynb b/bin/train/easy_colab.ipynb
@@ -74,7 +74,8 @@
"source": [
"!pip install git+https://github.com/sdatkinson/neural-amp-modeler.git@main\n",
"\n",
- "from nam.train.colab import run"
+ "from nam.train.colab import run\n",
+ "%load_ext tensorboard"
]
},
{
@@ -96,6 +97,7 @@
"metadata": {},
"outputs": [],
"source": [
+ "%tensorboard --logdir /content/lightning_logs\n",
"run(epochs=100)\n",
"# Psst! Curious how it's going?\n",
"# You can look under lightning_logs/version_0/checkpoints to see it saving its progress.\n",