首页 > 解决方案 > PyTorch 训练了哪些 ImageNet 类?

问题描述

PyTorch 允许您在此处下载用于分类的预训练模型,但目前尚不清楚如何从概率的输出张量返回到实际的类标签。有任何想法吗?

标签: pythonpytorch

解决方案



推荐阅读