-
I compile the torch(v1.4.0 cudatoolkit=10.2 cudnn=7.6.5) from source with the command "python setup install". And then load the model of shufflenet v2 0.5 with the compiled library(batchsize=8), the s…
-
Hello, everyone.
I'm new to Caffe2 and now working on my toy demo.
I wonder whether I can train my model with gpu and use(predict) it on cpu.
Any advice?
-
Hi,
I tried to include my own design of the network trained on proprietary data sets. I converted it from .caffemodel to caffe2 .pb model.
I have included all class names in "class.h" file and a…
-
I follow [this step](https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=prebuilt#install-with-gpu-support) but it shows the caffe2 does not have GPU support. And I check the lis…
-
How to convert multiple models to onnx format.
I have trained model which is having Gan model as well as encoder decoder models.
How do I convert all the models to onnx format
-
When building caffe2 I set my `CMAKE_INSTALL_PREFIX` to `~/venv3`, which correctly installed most of the files (headers, shared libs, binaries, cmake share files) to the correct locations in: `$PREFIX…
-
# Describe the bug
`caffe2/video/video_decoder.cc` use several parts of the FFMPEG API that have been [deprecated](https://ffmpeg.org/doxygen/4.0/deprecated.html) for a while, and that have been re…
gspr updated
10 months ago
-
I can succesfully save the PyTorch model into ONNX file now. However, The ONNX file can not be used by caffe2. This bug is about Upsample operator. @pk-g @houseroad
Acturally, I note that bugs for …
-
Hi,
We are trying to convert a pytorch model (originally written in 0.2 but it has been ported to pytorch 1.0)
to a caffe2 model, usign onnx framework to deploy on mobile.
We are are facing the…
-
aur package: _python-pytorch-cuda-git_ `ROCM=ON`
```sh
:: Downloading the latest sources for a devel package python-pytorch-cuda-git...
:: Starting the build:
==> Making package: python-pyto…