Skip to content

Instantly share code, notes, and snippets.

@YimianDai
Last active September 4, 2019 21:41
Show Gist options
  • Save YimianDai/7d31b979d47cffea18b9c7e2524fb75f to your computer and use it in GitHub Desktop.
Save YimianDai/7d31b979d47cffea18b9c7e2524fb75f to your computer and use it in GitHub Desktop.
MXNet Error

记录一些不容易找到问题就在哪的报错。

DataLoader:

  1. Check 一下是不是 Dataset 除了 img 和 label 之外还返回了 img name,导致 DataLoader 不对

Parameter was not initialized on context gpu(0). It was only initialized on [cpu(0)].

在 net.load_parameters 的语句之后添加 net.collect_params().reset_ctx(ctx = ctx)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment