-
Notifications
You must be signed in to change notification settings - Fork 9
Open
Description
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 call pytorch_model = gluon2pytorch(mod, [(batch_size[3], 3, 96, 96)], dst_dir=None, pytorch_module_name='Test')
Return: TypeError: 'Module' object is not callable
How can I solve this problem?How to convert module to be hybrid ?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels