From fbd2fd4ead22e0e24449ed33eae6d0ab611f2bf5 Mon Sep 17 00:00:00 2001
From: yh_cc <poemsmileyh@gmail.com>
Date: Sun, 26 Apr 2020 12:32:46 +0800
Subject: [PATCH] =?UTF-8?q?update=20DataSet=E4=BD=BF=E5=BE=97=E5=8F=AF?=
 =?UTF-8?q?=E4=BB=A5=E4=B8=8E=E4=B9=8B=E5=89=8D=E8=80=81=E7=89=88=E6=9C=AC?=
 =?UTF-8?q?=E7=9A=84DataSet=E5=85=BC=E5=AE=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fastNLP/core/dataset.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/fastNLP/core/dataset.py b/fastNLP/core/dataset.py
index a6c6cde6..464a6446 100644
--- a/fastNLP/core/dataset.py
+++ b/fastNLP/core/dataset.py
@@ -406,7 +406,18 @@ class DataSet(object):
 
             else:
                 raise ValueError("data only be dict or list type.")
-        self.collater = Collater()
+        self._collater = Collater()
+
+    @property
+    def collater(self):
+        if self._collater is None:
+            self._collater = Collater()
+        return self._collater
+
+    @collater.setter
+    def collater(self, value):
+        assert isinstance(value, Collater)
+        self._collater = value
 
     def __contains__(self, item):
         return item in self.field_arrays