Pytorch-多输出回归任务实战
# 注意以下几点 创建自己的类 MyDataset,继承自 torch.utils.data.Dataset,以便可以读取我们自己的数据集 在 Resnet50 的基础上,更改最后一层全连接层,添加 RelU 激活函数,Dropout,输出纬度更改为 [batchsz, 3],并将 dim=1 进行 Softmax 操作。 如果使用 GPU 设备,注意 model, loss_func, 和数据都要转移到 GPU 上,可以使用 cuda () 函数或 to (device)。 pytorch 的 dropout 的参数和 tf 的参数刚好相反 使用如下方式保存和加载模型: #...
more...







