gluon2pytorch

gluon2pytorch copied to clipboard

'Module' object is not callable

4

I loaded the model from .json and .params， sym, arg_params, aux_params = mx.model.load_checkpoint(prefix[3], epoch[3]) mod = mx.mod.Module(symbol=L106_Net96_v2("test"), context=mx.gpu(), data_names=['data'], label_names=['landmark_target']) mod.bind(for_training=False, data_shapes=[('data', (batch_size[3], 3, 96, 96))]) mod.set_params(arg_params, aux_params) When I...

guoguoguilai

Implement: - [ ] LSTM - [ ] GRU

gmalivenko