Add docstring for loss_type
no casting to str for loss_type, during headers creation