title: RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'
body: <p>I am trying to train <a href="https://github.com/yxlijun/Pelee.Pytorch" rel="nofollow noreferrer">PeleeNet pytorch</a> and got the following error</p>
  <p><a href="https://i.stack.imgur.com/6WUUS.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/6WUUS.png" alt="enter image description here" /></a></p>
  <p><code>train.py</code> <strong>line 80</strong>
  <a href="https://i.stack.imgur.com/KqP8b.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/KqP8b.png" alt="enter image description here" /></a></p>
  <p><code>pelee_voc</code> <strong>train configuration</strong>
  <a href="https://i.stack.imgur.com/iPt6o.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/iPt6o.png" alt="enter image description here" /></a></p>