MIC-DKFZ / nnUNet

Apache License 2.0
5.66k stars 1.71k forks source link

about v1 how to inference in cpu? #2467

Open qiaoqiangPro opened 1 month ago

qiaoqiangPro commented 1 month ago

I have understood how the nnunetv2 framework uses cpu inference, which is very convenient. But it seems that the nnunetv1 code can't set the device directly, do you know how to use cpu reasoning in the model inference phase and measure the inference time?