利用PyTorch實現VGG16

import torch import torch.nn as nn import torch.nn.functional as F class VGG16(nn.Module): def __init__(self): super(VGG16, self).__init__() # 3 * 224 * 224
相關文章
相關標籤/搜索