title: Training from checkpoint using Trainer
body: <p>I'm trying to continue training my model from a checkpoint but I get prompted this error:</p>
  <pre><code>RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)
  </code></pre>
  <p>I'm guessing the problem is from the data according to the error frames attached in the pic<a href="https://i.stack.imgur.com/VnOut.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/VnOut.png" alt="error frame" /></a></p>
  <p>But I already feed the data through the trainer object</p>
  <pre><code>trainer = AdapterTrainer(
      model=my_model,
      args=training_args,
      train_dataset=train,
      eval_dataset=test,
      compute_metrics=compute_metrics,
  )
  </code></pre>