It gives me :

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device cpu

basically I did :
model.eval()
model_cpu = model.module.to('cpu')
model_traced = torch.jit.trace(model_cpu, dummy_input)