XzJosh commited on
Commit
8768c3c
1 Parent(s): 7ed743a

Update config.yml

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_3500.pth"
106
  # 配置文件路径
 
100
  # 注意, “:” 后需要加空格
101
  webui:
102
  # 推理设备
103
+ device: "cpu"
104
  # 模型路径
105
  model: "models/G_3500.pth"
106
  # 配置文件路径