Roi Feng commited on
Commit
868935e
1 Parent(s): ac28faa

update config use CPU

Browse files
Files changed (1) hide show
  1. config.yml +1 -1
config.yml CHANGED
@@ -100,7 +100,7 @@ train_ms:
100
  # 注意, “:” 后需要加空格
101
  webui:
102
  # 推理设备
103
- device: "cuda"
104
  # 模型路径
105
  model: "models/G_18500.pth"
106
  # 配置文件路径
 
100
  # 注意, “:” 后需要加空格
101
  webui:
102
  # 推理设备
103
+ device: "cpu"
104
  # 模型路径
105
  model: "models/G_18500.pth"
106
  # 配置文件路径