Closed liangzz1991 closed 5 years ago
Solved. Replace every
torch::TensorFromBlob
to
at::from_blob
like
at::from_blob(weights_src, {length/4}).to(torch::kFloat32)
There will be 3 totally.
Originally posted by @zhengli233 in https://github.com/walktree/libtorch-yolov3/issues/12#issuecomment-495495994
Solved. Replace every
to
like
There will be 3 totally.
Originally posted by @zhengli233 in https://github.com/walktree/libtorch-yolov3/issues/12#issuecomment-495495994