Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- net = nn.Sequential(
- nn.Conv2d(1, 64, 3),
- nn.BatchNorm2d(64, momentum=1, affine=True),
- nn.ReLU(inplace=True),
- nn.MaxPool2d(2, 2),
- nn.Conv2d(64, 64, 3),
- nn.BatchNorm2d(64, momentum=1, affine=True),
- nn.ReLU(inplace=True),
- nn.MaxPool2d(2, 2),
- nn.Conv2d(64, 64, 3),
- nn.BatchNorm2d(64, momentum=1, affine=True),
- nn.ReLU(inplace=True),
- nn.MaxPool2d(2,2),
- Flatten(),
- nn.Linear(64, args.n_way)).to(device)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement