-
Notifications
You must be signed in to change notification settings - Fork 1.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
TF.JS on Node.js Error: Argument 'x' passed to 'slice' must be a Tensor, but got object #462
Comments
cc @caisq |
According to TF.js comment of the forum here I have updated the example code, in order to use I have then fixed the model.predict to use a $ node sentiment.js
2018-06-27 11:01:25.185649: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.2 AVX AVX2 FMA
(node:76671) Warning: N-API is an experimental feature and could change at any time.
============================
Hi there 👋. Looks like you are running TensorFlow.js in Node.js. To speed things up dramatically, install our node backend, which binds to TensorFlow C++, by running npm i @tensorflow/tfjs-node, or npm i @tensorflow/tfjs-node-gpu if you have CUDA. Then call require('tensorflow/tfjs-node'); (-gpu suffix for CUDA) at the start of your program. Visit https://github.com/tensorflow/tfjs-node for more details.
============================
indexFrom = 3
maxLen = 100
model_type cnn
vocabulary_size 20000
max_len 100
[ 'this', 'is', 'a', 'bad', 'day' ]
this is a bad day
Tensor {
isDisposedInternal: false,
size: 100,
shape: [ 1, 100 ],
dtype: 'float32',
strides: [ 100 ],
dataId: {},
id: 22,
rankType: '2' }
{ score: 0.018475884571671486, elapsed: 1594.1342370510101 } |
That warning is to be expected! |
@nsthorat ok so it is confirmed that I'm using Node.js so I can expect on linux to have the GPU working too, right? |
Yes, you can use @tensorflow/tfjs-node-gpu! |
@nsthorat thanks a lot then, that's crazy!!!! 💯 👍 🥇 |
TensorFlow.js version
"@tensorflow/tfjs": "^0.11.6",
"@tensorflow/tfjs-node": "^0.1.7"
Browser version
Node.js version
Describe the problem or feature request
I'm trying to convert a Tensorflow.js (ie the
tfjs
package) model example to the Node.js version (ie.tfjs-node
package).My import are the following:
This should be enough to load tfjs and tfjs-node bindings having tensorflow as default backend.
The code loads a pre-trained model built in shards (the tfjs model format), and it's as simple as:
I get this error while running:
Error: Argument 'x' passed to 'slice' must be a Tensor, but got object.
that means that my input object is not a Tensor object instance, even if I can clearly see in the logs that I have
a Tensor object instance when getting the tensor from the input buffer
const input = inputBuffer.toTensor();
that converts aTensorflowBuffer
to aTensor
object. This seems to not work properly in Node.js, while in the browser it works / or theassertion
type check does not work as expected when in Node.js.Code to reproduce the bug / link to feature request
Full code to reproduce the error: https://github.com/loretoparisi/tensorflow-node-examples/blob/master/sentiment/sentiment.js
The text was updated successfully, but these errors were encountered: