diff options
Diffstat (limited to 'utils/dataset.py')
| -rw-r--r-- | utils/dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/utils/dataset.py b/utils/dataset.py index bb4d762..ded9fd5 100644 --- a/utils/dataset.py +++ b/utils/dataset.py @@ -54,7 +54,7 @@ class CASIAB(data.Dataset): num_input_channels = 3, frame_height = 64, frame_width = 32) need about 22 GB of RAM. """ - super(CASIAB, self).__init__() + super().__init__() self._root_dir = root_dir self._is_train = is_train self._num_sampled_frames = num_sampled_frames |