X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=tinyae.py;h=b4f3aba8ed9fd3c1d23112818049551be2905e21;hb=5598ee31cbcd2ebbedeadbd66518f082c66aaaa9;hp=160878657efa7686c75ad8083b30fbfbb4e9d845;hpb=7e0d32924fda14c2ddfe23a63616eac33476d83c;p=pytorch.git

diff --git a/tinyae.py b/tinyae.py
index 1608786..b4f3aba 100755
--- a/tinyae.py
+++ b/tinyae.py
@@ -55,7 +55,7 @@ def log_string(s):
 
 class AutoEncoder(nn.Module):
     def __init__(self, nb_channels, embedding_dim):
-        super(AutoEncoder, self).__init__()
+        super().__init__()
 
         self.encoder = nn.Sequential(
             nn.Conv2d(1, nb_channels, kernel_size=5),  # to 24x24