def log_scalars()

in main_stica.py [0:0]

• 56 lines of code
• 5 McCabe index (conditional complexity)

def log_scalars(writer, loss, loss_crops, crop_losses, loss_av,
                counters, optimizer, batch_time, data_time, iteration):
    writer.add_scalar(
        f'pretrain/loss/iter',
        loss.item(),
        iteration
    )
    writer.add_scalar(
        f'pretrain/crop_loss/iter',
        loss_crops.item(),
        iteration
    )
    if counters[0] > 0:
        writer.add_scalar(
            f"pretrain/crop_loss/L-L/iter",
            crop_losses[0].item()/counters[0],
            iteration
        )
    if counters[1] > 0:
        writer.add_scalar(
            f"pretrain/crop_loss/S-L/iter",
            crop_losses[1].item() / counters[1],
            iteration
        )
    if counters[2] > 0:
        writer.add_scalar(
            f"pretrain/crop_loss/TL-TL/iter",
            crop_losses[2].item() / counters[2],
            iteration
        )
    if counters[3] > 0:
        writer.add_scalar(
            f"pretrain/crop_loss/TS-TL/iter",
            crop_losses[3].item() / counters[3],
            iteration
        )
    writer.add_scalar(
        f'pretrain/av_loss/iter',
        loss_av.item(),
        iteration
    )
    writer.add_scalar(
        f'pretrain/lr/iter',
        optimizer.param_groups[0]["lr"],
        iteration
    )
    writer.add_scalar(
        f'pretrain/batch_time/iter',
        batch_time.avg,
        iteration
    )
    writer.add_scalar(
        f'pretrain/data_time/iter',
        data_time.avg,
        iteration
    )