test/data/network2.py

import torch.nn as nn


class Net(nn.Module):

    def __init__(self):
        super(Net, self).__init__()
        self.linear = nn.Linear(10, 20)
        self.relu = nn.ReLU()