search

warmspringwinds / pytorch-cpp

Pytorch C++ Library
369 stars 68 forks source link

pretrained pytorch model #16

Open pharrellyhy opened 6 years ago

pharrellyhy commented 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?

pharrellyhy commented 6 years ago

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!

warmspringwinds commented 6 years ago

@pharrellyhy Dropout2d is not yet implemented but it's easy to do it though :)