pytorch-cpp icon indicating copy to clipboard operation
pytorch-cpp copied to clipboard
verified publisher icon warmspringwinds

pretrained pytorch model

Open pharrellyhy opened this issue 7 years ago • 2 comments

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 avatar Jun 13 '18 10:06 pharrellyhy

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 avatar Jun 13 '18 11:06 pharrellyhy

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

warmspringwinds avatar Jun 20 '18 14:06 warmspringwinds