Function createDummyModel

tfjs-layers/src/engine/training_test.ts:2221–2229

(inputSize: number)

describeMathGPU('LayersModel.fit: yieldEvery', () => {
  function createDummyModel(inputSize: number): tfl.LayersModel {
    const model = tfl.sequential();
    const layerSize = 10;
    model.add(tfl.layers.dense(
        {units: layerSize, inputShape: [inputSize], activation: 'relu'}));
    model.add(tfl.layers.dense({units: 1}));
    model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
    return model;
  }

  it('auto: 1 batch per epoch; 5 epochs', async () => {
    const wait = DEFAULT_YIELD_EVERY_MS;