20 lines
569 B
Python
20 lines
569 B
Python
import torch.nn as nn
|
|
|
|
mai = nn.Sequential(
|
|
nn.Conv2d(in_channels=3, out_channels=16, kernel_size=3, padding=1),
|
|
nn.ReLU(),
|
|
nn.MaxPool2d(kernel_size=2, stride=2),
|
|
nn.Conv2d(in_channels=16, out_channels=32, kernel_size=3, padding=1),
|
|
nn.ReLU(),
|
|
nn.MaxPool2d(kernel_size=2, stride=2),
|
|
nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3, padding=1),
|
|
nn.ReLU(),
|
|
nn.MaxPool2d(kernel_size=2, stride=2),
|
|
nn.Flatten(),
|
|
nn.Linear(25 * 25 * 64, 120),
|
|
nn.ReLU(),
|
|
nn.Linear(120, 30),
|
|
nn.ReLU(),
|
|
nn.Linear(30, 1),
|
|
)
|