Hi!
I found some strange things. The exceptions can be ignored. As a sanity check I ran the example code with resnet18 of pytorch and this worked fine. When I train my model on the cpu the output of the forward function in C++ is fine. But when I train it on the GPU it is ‘None’, both on the cpu and on the gpu. In python this GPU trained model works fine both on the cpu and gpu. The cuda versions are the same ( 10 ).
Can anybody help me figuring out what is wrong?
Greetings,
Roos