Tensorflow JS模型拟合立即完成，无需执行任何操作

Question

嗨，我正在尝试建立一个卷积神经网络，但我不能训练它。

代码如下：

model = tf.sequential();
model.add(tf.layers.conv2d({
    inputShape: [48, 48, 1],
    kernelSize: FILTER_SIZE,
    filters: 64,
    dataFormat: "channelsLast",
    activation: ActFunc.RELU
}));
model.add(tf.layers.maxPooling2d(maxPoolConf));
model.add(tf.layers.conv2d({
    kernelSize: FILTER_SIZE,
    filters: 128,
    dataFormat: "channelsLast",
    activation: ActFunc.RELU
}));
model.add(tf.layers.maxPooling2d(maxPoolConf));
model.add(tf.layers.conv2d({
    kernelSize: FILTER_SIZE,
    filters: 256,
    dataFormat: "channelsLast",
    activation: ActFunc.RELU
}));
model.add(tf.layers.maxPooling2d(maxPoolConf));
model.add(tf.layers.conv2d({
    kernelSize: FILTER_SIZE,
    filters: 512,
    dataFormat: "channelsLast",
    activation: ActFunc.RELU
}));
model.add(tf.layers.maxPooling2d(maxPoolConf));
model.add(tf.layers.flatten());
model.add(tf.layers.dense({units: 128, activation: 'relu'}));
model.add(tf.layers.dense({units: 256, activation: 'relu'}));
model.add(tf.layers.dense({units: 512, activation: 'relu'}));
model.add(tf.layers.dense({units: 1024, activation: 'relu'}));
model.add(tf.layers.dense({
    units: 7,
    activation: 'softmax'
}));
model.compile({
    optimizer: 'adam',
    loss: 'categoricalCrossentropy',
    metrics: ['accuracy', 'categoricalCrossentropy']
});

let image_tensor = tf.tensor4d(training_data.getInputData(), [training_data.length, 48, 48, 1]);
let correct_prediction_tensor = tf.tensor2d(training_data.getLabels(), [training_data.length, 7]);

const history = await model.fit(image_tensor, correct_prediction_tensor,
    {
        batchSize: 128,
        epochs: 10,
        shuffle: true,
        callbacks: {
            onEpochEnd: (epoch, logs) => {
                // Plot the loss and accuracy values at the end of every training epoch.
                console.log(epoch, logs);
            },
            onTrainStart: console.log("Starting Training..."),
            onTrainEnd: console.log("Training Finished!"),
        }
    });

当我运行这段代码时，它会打印“开始训练...”紧接着“训练结束了！”(它甚至不训练模型)，然后我的GPU有100%的负载，直到我关闭选项卡。该怎么办呢。

输入数据为48x48图像。

training_data.getInputData()返回一个平面数组，其中包含每个图像的像素数据，而training_data.getLabels()返回一个包含标签数据的平面数组。

Answer 1

model.fit()是一种异步方法。您应该使用await或then。

例如，使用await (从ES2017开始)：

  const history = await model.fit(image_tensor, correct_prediction_tensor,
    {
        batchSize: 128,
        epochs: 10,
        shuffle: true,
        callbacks: {
            onEpochEnd: (epoch, logs) => {
                // Plot the loss and accuracy values at the end of every training epoch.
                console.log(epoch, logs);
            },
            onTrainStart: console.log("Starting Training..."),
            onTrainEnd: console.log("Training Finished!"),
        }
    });

或使用then

  model.fit(image_tensor, correct_prediction_tensor,
    {
        batchSize: 128,
        epochs: 10,
        shuffle: true,
        callbacks: {
            onEpochEnd: (epoch, logs) => {
                // Plot the loss and accuracy values at the end of every training epoch.
                console.log(epoch, logs);
            },
            onTrainStart: console.log("Starting Training..."),
            onTrainEnd: console.log("Training Finished!"),
        }
    }).then(history => {
      console.log('history:', history);
    });