From 4e8f6845c2c42a164fb4b689f6206fb46d18bd88 Mon Sep 17 00:00:00 2001
From: castle <1090414006@qq.com>
Date: Fri, 13 Aug 2021 12:19:33 +0800
Subject: [PATCH] fix dataloader bug

---
 utils/data_loaders.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/data_loaders.py b/utils/data_loaders.py
index 9ae14a9..21edda6 100644
--- a/utils/data_loaders.py
+++ b/utils/data_loaders.py
@@ -89,6 +89,7 @@ def get_dataset(self, subset):
         file_list = self._get_file_list(self.cfg, self._get_subset(subset), n_renderings)
         transforms = self._get_transforms(self.cfg, subset)
         return Dataset({
+            'n_renderings': n_renderings,
             'required_items': ['partial_cloud', 'gtcloud'],
             'shuffle': subset == DatasetSubset.TRAIN
         }, file_list, transforms)