hi Ptrblck,
after much head scratching, i think the following seems to have solved it. However i am intrigued on the difference between this and your solution, which would be more stable?

test_input=test_input.cuda()

Chaslie