block13.csharp.cs

// create the batch to train on
var trainingBatch = StyleTransfer.CreateBatch(lossFunction, labels);

// train the model
Console.WriteLine("Training the model...");
var numEpochs = 300;
for (int i = 0; i < numEpochs; i++)
{
    trainer.TrainMinibatch(trainingBatch, true, NetUtil.CurrentDevice);
    if (i % 50 == 0)
        Console.WriteLine($"epoch {i}, training loss = {trainer.PreviousMinibatchLossAverage()}");
}

// the rest of the code goes here...