How to load_model correctly

lucidrains / lambda-networks

Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute

MIT License

1.53k stars 157 forks source link

Open JNaranjo-Alcazar opened 2 years ago

JNaranjo-Alcazar commented 2 years ago

Hi everyone, I am struggling when loading this model saved in .h5 file.

How is the correct way to load this network? If I use custom_objects I get init() got an unexpected keyword argument 'name'