上传者: 42166105
|
上传时间: 2022-03-08 19:03:23
|
文件大小: 5KB
|
文件类型: -
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