Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #use this one for 4d
- class Net(nn.Module):
- def __init__(self, count1, count2, fs):
- super(Net, self).__init__()
- self.conv1 = nn.Conv2d(3, count1, fs)
- self.pool = nn.MaxPool2d(2, 2)
- self.conv2 = nn.Conv2d(count1, count2, fs)
- self.fc1 = nn.Linear(count2 * fs * fs, 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 = x.view(-1, 16 * 3 * 3)
- # x = x.view(-1, 32 * 3 * 3)
- # x = x.view(-1, 64 * 3 * 3)
- # x = x.view(-1, 128 * 3 * 3)
- # x = x.view(-1, 256 * 3 * 3)
- # x = x.view(-1, 512 * 3 * 3)
- # x = x.view(-1, 1024 * 3 * 3)
- x = F.relu(self.fc1(x))
- x = F.relu(self.fc2(x))
- x = self.fc3(x)
- return x
- #net = Net()
- #net = Net(6, 16, 5)
- #net = Net(12, 32, 5)
- #net = Net(24, 64, 5)
- #net = Net(48, 128, 5)
- #net = Net(96, 256, 5)
- #net = Net(192, 512, 5)
- #net = Net(384, 1024, 5)
- #use for 4d
- net = Net(6, 16, 3)
- #net = Net(12, 32, 3)
- #net = Net(24, 64, 3)
- #net = Net(48, 128, 3)
- #net = Net(96, 256, 3)
- #net = Net(192, 512, 3)
- #net = Net(384, 1024, 3)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement