diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-17 09:22:10 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-17 09:22:10 +0200 |
| commit | 88976997040c542f874cdaf1d67bb7c47e96a536 (patch) | |
| tree | 08fc9f53f96b8f8ec1adf0f4066b3018d507b3c9 | |
| parent | 9467998d18246acb60d76349826b2e8ebaedfe9c (diff) | |
args
| -rw-r--r-- | data/recursive_dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 87f331c..728bfba 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -35,7 +35,7 @@ class RecursiveDataset(BaseDataset): return input_dict def __len__(self): - return len(self.A_paths) + return self.dataset_size def name(self): return 'RecursiveDataset'
\ No newline at end of file |