PyTorch - 多任务网络之年龄与性别预测数据集类的构建多任务网络的搭建与训练
目录数据集类的构建torch.utils.data.Datasetimplementation导入包\__init__()\__len__()\__getitem__()检验数据集类手动提取数据DataLoader多任务网络的搭建与训练导入包检查是否可使用GPU多任务网络的搭建设置网络(\__init__())forward()训练过程数据集类的构建本实例使用的是UTKFace数据集,包含了两万多张不同种族的不同年龄的人脸图片torch.utils.data.Dataset是一个抽象类, 自定义