From 5329614018a52d851b2e8cfced7a18df88a5dab7 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Tue, 3 May 2022 09:36:17 +0000
Subject: [PATCH] small

---
 fastNLP/core/dataloaders/jittor_dataloader/fdl.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fastNLP/core/dataloaders/jittor_dataloader/fdl.py b/fastNLP/core/dataloaders/jittor_dataloader/fdl.py
index 9b67629e..507073a4 100644
--- a/fastNLP/core/dataloaders/jittor_dataloader/fdl.py
+++ b/fastNLP/core/dataloaders/jittor_dataloader/fdl.py
@@ -91,6 +91,7 @@ class JittorDataLoader:
             self.dataset.dataset.set_attrs(batch_size=1)
         # 用户提供了 collate_fn，则会自动代替 jittor 提供 collate_batch 函数
         # self._collate_fn = _collate_fn
+        self.cur_batch_indices = None
 
     def __iter__(self):
         # TODO 第一次迭代后不能设置collate_fn，设置是无效的