xiaochus / MobileNetV3

A Keras implementation of MobileNetV3.
MIT License
236 stars 83 forks source link

TypeError: _hard_swish() missing 1 required positional argument: 'x' #17

Open Cai-YingJie opened 3 years ago

Cai-YingJie commented 3 years ago

This error occurred when I loaded the model。 h5_model = load_model(weight_file_path, custom_objects={'_hard_swish':_hard_swish, '_relu6':_relu6, '_return_activation':_return_activation, '_conv_block':_conv_block, '_bottleneck':_bottleneck, 'build':build})

h5_to_pb(h5_model,output_dir = output_dir,model_name = output_graph_name)