diff --git a/main.ipynb b/main.ipynb index 3d2d095dc242ad1d41296300fac97bfd9ab77664..56f9685c6a43f3e76216707292125aa41a16b5a1 100644 --- a/main.ipynb +++ b/main.ipynb @@ -303,7 +303,7 @@ "metadata": {}, "outputs": [], "source": [ - "NAME = 'Baseline_VAEP_short_512_10=warmup_begin'\n", + "NAME = 'Baseline_VAE_short_512_10=warmup_begin'\n", "log_dir = os.path.abspath(\n", " \"logs/fit/{}_{}\".format(NAME, datetime.now().strftime(\"%Y%m%d-%H%M%S\"))\n", ")\n",