diff --git a/beginner_source/basics/buildmodel_tutorial.py b/beginner_source/basics/buildmodel_tutorial.py index da1974fcb0e..d2c0572d1ff 100644 --- a/beginner_source/basics/buildmodel_tutorial.py +++ b/beginner_source/basics/buildmodel_tutorial.py @@ -49,7 +49,7 @@ class NeuralNetwork(nn.Module): def __init__(self): - super(NeuralNetwork, self).__init__() + super().__init__() self.flatten = nn.Flatten() self.linear_relu_stack = nn.Sequential( nn.Linear(28*28, 512),