Various changes

[vitkl]

• [ ] Delete switch from eval to train mode in here https://github.com/dissatisfaction-ai/scHierarchy/blob/main/schierarchy/logistic/_logistic_model.py#L411