pytorch-cpp
pytorch-cpp copied to clipboard
warmspringwinds
Pytorch C++ Library
Hi! Does pytorch-cpp support any conversion from customized models built in pytorch, or does it only support standard models (e.g. vgg, res-net) and your own segmentation models? Big thanks!
The error occurred at runtime while trying to run this example: https://github.com/goldsborough/examples/blob/cpp/cpp/mnist/mnist.cpp Stack trace: ``` terminate called after throwing an instance of 'c10::Error' what(): input.ndimension() == 4 ASSERT FAILED at...
I have a tensor A shape [1,256,15,20] and B shape [1,1,15,20], when I use A.div(B), it error. the error is `terminate called after throwing an instance of 'std::runtime_error' what(): invalid...
When I run the convert_weights.ipynb with my own model which is a pretrained Dual Path Network(DPN)：  A mistake happend ：   Could you please tell me why this...
I converted a custom resnet18 model weights to h5 and loaded to the cpp version model which l created based on your sample code. I also inspected the weights before...
How to convert Type of Aten? from kByte to kFloat?
Conversion from float std::vector to Aten Tensor and vice versa?
Where Are you get this function ? What is version of hdf5 version? I use 1.10.1 And I can't find get_hdf5_file_keys() fucntion
Hi, I've a pretrained model which is saved as a pth.tar file, so is it possible to use pytorch-cpp to inference?