X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=tinyae.py;h=70484f151c866d57e1ee8cd1b25c2b3d5b87d483;hp=160878657efa7686c75ad8083b30fbfbb4e9d845;hb=HEAD;hpb=05b9b133a45ac9bd5abe6f8b6d29095f9c82797a

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