MiniTorch / project / train_utils.py
train_utils.py
Raw
def default_log_fn(epoch, total_loss, correct, losses):
    print("Epoch ", epoch, " loss ", total_loss, "correct", correct)


def default_plot_fn(fig, losses):
    pass