关于 YOLO v10 使用过程中的小问题求教一下各位 v 友

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 117 天前的主题，其中的信息可能已经有所发展或是发生改变。

关键代码如下：

device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(f"Using device: {device}")

# 加载预训练的 YOLOv10n 模型并将其移到 GPU （如果可用）
model = YOLO("yolov10n.pt").to(device)

部分输出如下：

Using device: cpu

明明都没有使用 GPU ，但是末尾处加 to(device)后识别准确率和速度大幅上升，不加则非常卡而且无法正常识别，今天刚刚上手，不是很清楚怎么回事

4 条回复 • 2024-08-02 09:35:16 +08:00

ysc3839

117 天前 via Android

改成.eval()呢？

WilliamColton

117 天前

@ysc3839 #1 好像直接开始训练了？但是直接 to(device) 或者没有 to(device) 是不会有这个过程的

WilliamColton

117 天前

@ysc3839 #1 然后每次训练不久就会卡死，调试之后发现好像是被“锁”住了

ysc3839

116 天前 via Android

@WilliamColton .eval() 不会开始训练吧？
https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module.eval
等价于 self.train(False)