search

ntedgi / node-efficientnet

tensorflowJS implementation of EfficientNet 🚀
https://www.npmjs.com/package/node-efficientnet
MIT License
257 stars 37 forks source link

add support to load your own model using EfficientNetCheckPointFactory #6

Closed ntedgi closed 3 years ago

ntedgi commented 3 years ago

for now, we only support getting checkpoint as a parameter and not allowing the user to load is own model

add support on EfficientNetCheckPointFactory