Hey, I’ll just reply quickly now but the second posting I did is running for me. Perhaps you mixed some of my code with your own? The error message you are getting is because the model and input are not on the same device, meaning they arent both on either the CPU or the GPU. Try adding the model = model.to(device) as in your original code