随笔分类 -  pytorch

摘要：本文是《小土堆》课程的笔记之dataset类实战 from torch.utils.data import Dataset from PIL import Image import os class MyData(Dataset): def __init__(self, root_dir,label 阅读全文