DAY29 demo文件程式問題 - Cupoy

2020/01/13 下午 06:29

電腦視覺深度學習討論版

蕭淳

觀看數：2

回答數：1

收藏數：0

cvdl-1

cvdl-1-d29

請問這個問題該怎麼解決

回答列表

• 2020/01/13 下午 10:50
  楊哲寧
  贊同數：1
  不贊同數：0
  留言數：0

  您好，不好意思這裡是程式碼上的疏忽，我們可以將model 轉換為cuda即可：

  net.cuda()     # wrap tensor in Variable

  或是我們把程式碼中這段移除，用cpu做inference:

  if torch.cuda.is_available():

    xx = xx.cuda()