I tried to run your code but it stucked at line 13 of Trainer.py self.network = self.network.to(device), it said RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at ..\aten\src\THC\THCGeneral.cpp:50
Any special config? I'm currently using pytorch 1.4.0
I am trying to convert the DeePixBiS model, but on the inference I get a completely different result in contrast to the original implementation of the model. This is how I convert from onnx to rknn https://pastebin.com/cNJvijCe , and this is the inference code https://pastebin.com/jy6Y6EeE . Could you tell me what I 'm doing wrong ?