From 2aadc23c7c7e92972286a16e18c2d190448b1c5c Mon Sep 17 00:00:00 2001
From: Conight <xfddwhh@gmail.com>
Date: Wed, 22 Jun 2022 13:17:06 +0800
Subject: [PATCH] Fix train decoder config example (#160)

---
 configs/train_decoder_config.example.json | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/train_decoder_config.example.json b/configs/train_decoder_config.example.json
index f08758e..d618579 100644
--- a/configs/train_decoder_config.example.json
+++ b/configs/train_decoder_config.example.json
@@ -15,7 +15,7 @@
         "channels": 3,
         "timesteps": 1000,
         "loss_type": "l2",
-        "beta_schedule": "cosine",
+        "beta_schedule": ["cosine"],
         "learned_variance": true
     },
     "data": {