diff options
| -rw-r--r-- | models/layers.py | 2 | ||||
| -rw-r--r-- | utils/dataset.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/models/layers.py b/models/layers.py index c69ae07..a9f04b3 100644 --- a/models/layers.py +++ b/models/layers.py @@ -154,7 +154,7 @@ class BasicConv1d(nn.Module): kernel_size: Union[int, tuple[int]], **kwargs ): - super(BasicConv1d, self).__init__() + super().__init__() self.conv = nn.Conv1d(in_channels, out_channels, kernel_size, bias=False, **kwargs) 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 |