weights_for(model_name: str)Return the WeightsEnum registered for model_name.
Parameters
model_namestrFactory name, e.g.
"resnet_18" — the same string passed
to register_weights at module import time.Returns
type[WeightsEnum] or NoneThe registered enum class, or None when the model has
no pretrained weights declared yet.
Examples
>>> import lucid.weights as W
>>> W.weights_for("resnet_18")
<enum 'ResNet18Weights'>