ntedgi / node-efficientnet

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

Configurable download address #27

Closed luoye-fe closed 3 years ago

luoye-fe commented 3 years ago

Is your feature request related to a problem? Please describe. You know, downloading github assets in mainland China is particularly slow~

Describe the solution you'd like Maybe, allow to specify local model dir? Such as:

EfficientNetCheckPointFactory.create(EfficientNetCheckPoint.B7, {
  localModelRootDir: path.join(__dirname, 'tf-model'),
});

Describe alternatives you've considered

Additional context Add any other context or screenshots about the feature request here.

ntedgi commented 3 years ago

should be enabled as part of #6

luoye-fe commented 3 years ago

28