• Vasilis Vryniotis's avatar
    Refactor the `get_weights` API (#5006) · 3d8723d5
    Vasilis Vryniotis authored
    * Change the `default` weights mechanism to sue Enum aliases.
    
    * Change `get_weights` to work with full Enum names and make it public.
    
    * Applying improvements from code review.
    3d8723d5
train.py 20.9 KB