xiaochus / MobileNetV3

A Keras implementation of MobileNetV3.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

Cai-YingJie opened this issue · comments

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)