ManTraNet-pytorch

ManTraNet-pytorch copied to clipboard

Some questions

4

Hi! I would like to ask if the parameters of the pre-trained model you provided were converted from the original author's tensorflow version? Then can you provide the relevant training...

kerryhhh

Exporting model to ONNX format fails

2

Exporting model in ONNX format following [the process described in PyTorch's documentation](https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html) fails. Below I attach returned error message: ``` /home/dkarageo/development/ManTraNet-pytorch/venv/lib/python3.9/site-packages/torch/cuda/__init__.py:82: UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you...

dkarageo

the network output will become nan

8

  I am very grateful to the author for the pytorch version code. During my training, I found that the network output will become nan. Has the author encountered...

15066952332