Open pharrellyhy opened 6 years ago
Hi,
I've a pretrained model which is saved as a pth.tar file, so is it possible to use pytorch-cpp to inference?
Also, I noticed that you didn't implement nn.Dropout2d for now while I have this op in my graph. So, any advice on this? Thanks!
@pharrellyhy Dropout2d is not yet implemented but it's easy to do it though :)
Hi,
I've a pretrained model which is saved as a pth.tar file, so is it possible to use pytorch-cpp to inference?