All Questions
Tagged with libtorch machine-learning
3 questions
0
votes
0
answers
104
views
torch.jit.script Error for nn.Module attributes in the __init__: Module has no attribute
I am trying to use torch.jit.script to script the demucs audio separation ML model.
Ultimately, I'm trying to make the model compatible with coreml to use it in an iOS app
RuntimeError:
Module '...
- 9
2
votes
0
answers
484
views
How to retrieve x and y coordinates in torch tensor?
I'm working with a mobilenetv2 model. So I'm working with detection and I'm able to get the score (how close an object resembles something on the training set list) and the object name.
I just need to ...
- 21
3
votes
2
answers
1k
views
slow Inference time for Neural Net
I have written a simple Fully connected Neural Network in Pytorch. I saved the model and loaded it in C++ using LibTorch but my inference time is pretty slow for my application field. Inference time ...
- 31