10种猴子
Kaggle链接: ://www.kaggle.com/slothkong/10-monkey-species/notebooks datasetId 10449
基本模型架构:
self.layer1 = torch.nn.Sequential(torch.nn.Conv2d(1, 28, kernel_size=5),
torch.nn.ReLU(),
torch.nn.MaxPool2d(kernel_size=2))
self.layer2 = torch.nn.Sequential(torch.nn.Conv2d(28, 10, kernel_size=3),
torch.nn.ReLU(),
torch.nn.MaxPool2d(kernel_size=2))
self.dropout1 = t
1