You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
There are a few bugs in DJL right now when you try to use multiple engines.
For example if we use MXNet and TensorFlow engine together. If I set -Dai.djl.default_engine=MXNet, and call a TfEngine or TfModelZoo method, MxEngine and MxModelZoo is actually returned.
TfEngine.getInstance() will return Default Engine instead of TfEngine.
TfModelZoo.RESNET.loadModel() will return MxModel.RESNET if default engine is MXNet, will return PyModel.RESNET if default engine is Pytorch. But user already specified to use TfModelZoo
In Criteria.builder() the .optEngine("TensorFlow") option is not used by ModelZoo during loading model.
Right now the 2 ways to work with multiple engines are:
manually switch engine by setting system property ai.djl.default_engine back and forth
use newInstance with engineName will return the correct implementation under that engine: Model tfModel = Model.newInstance(Device.defaultDevice(), "TensorFlow")
The text was updated successfully, but these errors were encountered:
There are a few bugs in DJL right now when you try to use multiple engines.
For example if we use MXNet and TensorFlow engine together. If I set -Dai.djl.default_engine=MXNet, and call a TfEngine or TfModelZoo method, MxEngine and MxModelZoo is actually returned.
TfEngine.getInstance()
will return Default Engine instead of TfEngine.TfModelZoo.RESNET.loadModel()
will return MxModel.RESNET if default engine is MXNet, will return PyModel.RESNET if default engine is Pytorch. But user already specified to use TfModelZooIn
Criteria.builder()
the.optEngine("TensorFlow")
option is not used by ModelZoo during loading model.Right now the 2 ways to work with multiple engines are:
ai.djl.default_engine
back and forthnewInstance
with engineName will return the correct implementation under that engine:Model tfModel = Model.newInstance(Device.defaultDevice(), "TensorFlow")
The text was updated successfully, but these errors were encountered: