Do we want to keep support that option? I suggest to use the `name` variable to pick the right wights in the new API ([API ref](https://pytorch.org/vision/stable/models.html) ) _Originally posted by @SagiPolaczek in https://github.com/BiomedSciAI/fuse-med-ml/pull/291#discussion_r1139188045_