莫煩pytorch(10)——CNN

import torchvision用來生成圖片,視頻數據集,和一些流行的模型類和預訓練模型。 1.生成數據集 import torch import torch.nn as nn import torch.utils.data as Data import torchvision # 數據庫模塊 import matplotlib.pyplot as plt torch.manual
相關文章
相關標籤/搜索