ptrblck:

If @rasbt’s suggestions don’t help, could you try to assign the model to the CPU version?


You mean
model = model.cpu()

? That’s how I typically do it out of habit, but I somewhere read that it should be an in-place operation. The assignment certainly doesn’t hurt though.