Make tfjs use all 4 cpu cores

TensorFlow General Discussion
,
1

Hey guys! I am trying to train a layered model with tensorflow.js but the training only uses 1 core instead of all the/4 cores
image

await model.fit(xTrain, yTrain, {
        batchSize: 32,
        validationSplit: 0.1,
        shuffle: true,
        epochs: 500,
        callbacks: { onBatchEnd }

here is how i train the model^
Can someone tell me how to add the tags as i have to add the tfjs tag

3

Welcome to the forum and thanks for posting!

So from what I understand TFJS Node uses the eager mode TF C++ API behind the scenes for model training where by only certain ops are parallelizable. If those ops are not being used then you will see the single core usage as shown in your screenshot. I believe TF Python uses eager mode by default too if I remember correctly to avoid issues.

It should be noted however that when using a saved model in TFJS Node, the act of performing inference however can utilize inter-op parallelization as graph mode is used so saved model usage is still very fast for models that support parallel op execution in Node JS too. The above only applies to the training.

For more details on graph/eager mode for execution check this interesting blog post (even though this is about Python I believe similar rules apply here too):

4

Is the multithread only available with the wasm backend?