From 71cab7230d2853450fbd604e6e45c7d531e080f1 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Thu, 4 Jun 2020 22:49:19 +0200
Subject: [PATCH] Implemented KL and MMD warmup on SEQ2SEQ_VAEP in models.py
---
main.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.ipynb b/main.ipynb
index 3d2d095d..56f9685c 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",
--
GitLab