Warning: Encountered known unsupported method torch.Tensor.new_tensor

Question

FX-STAR opened this issue 3 years ago · 3 comments

Cascade RCNN convert, any advice ? Thanks

Answer 1 · 2021-02-07T01:22:56.000Z

Just ignore it.

Answer 2 · 2021-02-07T07:49:38.000Z

thank you for your reply，but how to choose device dynamic

device='cuda:1'
trt_model = init_detector(trt_model_path).to(device)

not work

Answer 3 · 2021-02-07T09:48:11.000Z

The model can only work on the device you convert it.
If you want to use another device, please set the enviroment variable CUDA_VISIBLE_DEVICES