private void TrainMinibatch()
{
UnityEngine.Profiling.Profiler.BeginSample("TrainMiniBatch");
float avgTrainCost = 0f;
DataManager.GetBatch(_batch, DataManager.Train, ref _rng);
var handle = NeuralJobs.ZeroGradients(_gradientsAvg);
for (int i = 0; i < _batch.Length; i++)
{
handle = DataManager.CopyInput(_net.Inputs, DataManager.Train, _batch[i], handle);
handle = NeuralJobs.ForwardPass(_net, handle);
int lbl = (int)DataManager.Train.Labels[_batch[i]];
handle.Complete();
NeuralMath.ClassToOneHot(lbl, _targetOutputs); // Todo: job
handle = NeuralJobs.BackwardsPass(_net, _gradients, _targetOutputs, handle);
handle = NeuralJobs.AddGradients(_gradients, _gradientsAvg, handle);
handle.Complete();
// Todo: backwards pass logic now does this, don't redo, just check
NeuralMath.Subtract(_targetOutputs, _net.Last.Outputs, _dCdO);
float cost = NeuralMath.Cost(_dCdO);
avgTrainCost += cost;
}
// Update weights and biases according to averaged gradient and learning rate
_rate = 3.0f / (float)BatchSize;
handle = NeuralJobs.UpdateParameters(_net, _gradientsAvg, _rate, handle);
handle.Complete(); // Todo: Is this one needed?
_batchCount++;
avgTrainCost /= (float)BatchSize;
_trainingLoss = (float)System.Math.Round(avgTrainCost, 6);
UnityEngine.Profiling.Profiler.EndSample();
}
