Tomash:

    mae, f1 = validate(val_loader, model, epoch, device)


If I understand it correctly, simply passing train_loader in place of val_loader should give you the mae and f1 for training dataset.