数据集

Dataset

DataLoader

一般来说用数据集训练都是这个流程:

# 创建Dateset(可以自定义)
    dataset = face_dataset # Dataset部分自定义过的face_dataset
# Dataset传递给DataLoader
    dataloader = torch.utils.data.DataLoader(dataset,batch_size=64,shuffle=False,num_workers=8)
# DataLoader迭代产生训练数据提供给模型
    for i in range(epoch):
        for index,(img,label) in enumerate(dataloader):
            pass

模型保存与加载

.pt .pth .bin .onnx格式

Numerical Optimization «
Prev «
» Generating Music with GANs
» Next