- Torch to Theano Converter
- no implementation for Tensorflow and other framework
- Converter
from ml_inference.modeling import *
import numpy as np
import torch
hooking_dummy = torch.Tensor(np.random.rand(3,64,64))
weight_parser(dnn_model, 'torch', hooking_dummy)2.restore model.
from ml_inference.modeling import *
model = NeuralNet('weights.h5')- The light package using theano with Scikit-Learn can upload to AWS Lambda.
- It is slow than Pytorch as Theano need not setup G++ environment.
- This library can't support Tensorflow or MXNet.
Hyunjune Kim - email is '4u_olion@naver.com' , You can call me Jey!
Kyungyong Lee - my professor is him, an assistant professor in KOOKMIN University.
