problem loading model with torch>=2.6

[mdciri]

it is not possible to do a training with torch>=2.6 because the torch.load() changed and does not load the model.