经典卷积神经网络---AlexNet网络
AlexNet网络结构及代码
AlexNet网络结构
AlexNet网络构建代码
模型构建
import torch.nn as nn
import torch
class AlexNet(nn.Module):
def __init__(self, num_classes=1000, init_weights=False):
super(AlexNet, self).__init__()
self.features = nn.Sequential(
原创
2022-05-27 17:42:39 ·
278 阅读 ·
0 评论