diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-31 10:57:35 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-31 10:57:35 +0800 |
| commit | 0d307facb16236174c302dc7714f7f20f434a4a6 (patch) | |
| tree | 9d4c76cc1e55a129bea44af1f11e2fd234ae73f5 | |
| parent | 07de5a9797a4f53779fe43c14ee51bdc4e248463 (diff) | |
Make super class constructor revoke consistent
| -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 |