How to load a pytorch model or an onnx model into this framework?

hermosayhl / CNN

C++ inplementation of CNN(Cnnvolutional Neural Network) for image classification

82 stars 30 forks source link

How to load a pytorch model or an onnx model into this framework? #1

Closed jawaechan closed 2 years ago

jawaechan commented 2 years ago

How to load a pytorch model or an onnx model into this framework? Thanks

hermosayhl commented 2 years ago

Sorry, the framework is just a demo for the simplest inplementation of convolutional neural network, only to verify the forward and backward processing for myself.

Thus, the code is not consistent with any other deep learning architectures or inference engine.