Removed the no use SimpleNetwork.

examples/hello-world/hello-pt-resnet/src/simple_network.py

@@ -13,33 +13,11 @@
 # limitations under the License.
 from typing import Optional, Any
 
-import torch
-import torch.nn as nn
-import torch.nn.functional as F
 from torchvision.models import ResNet
 from torchvision.models._utils import _ovewrite_named_param
 from torchvision.models.resnet import BasicBlock, ResNet18_Weights
 
 
-class SimpleNetwork(nn.Module):
-    def __init__(self):
-        super(SimpleNetwork, self).__init__()
-        self.conv1 = nn.Conv2d(3, 6, 5)
-        self.pool = nn.MaxPool2d(2, 2)
-        self.conv2 = nn.Conv2d(6, 16, 5)
-        self.fc1 = nn.Linear(16 * 5 * 5, 120)
-        self.fc2 = nn.Linear(120, 84)
-        self.fc3 = nn.Linear(84, 10)
-
-    def forward(self, x):
-        x = self.pool(F.relu(self.conv1(x)))
-        x = self.pool(F.relu(self.conv2(x)))
-        x = torch.flatten(x, 1)  # flatten all dimensions except batch
-        x = F.relu(self.fc1(x))
-        x = F.relu(self.fc2(x))
-        x = self.fc3(x)
-        return x
-
 class Resnet18(ResNet):
 
     def __init__(self, num_classes, weights: Optional[ResNet18_Weights] = None, progress: bool = True, **kwargs: Any):