• Peizhao Zhang's avatar
    Refactor for get_optimizer_param_groups. · 2dc3bc02
    Peizhao Zhang authored
    Summary:
    Refactor for get_optimizer_param_groups.
    * Split `get_default_optimizer_params()` into multiple functions:
      * `get_optimizer_param_groups_default()`
      * `get_optimizer_param_groups_lr()`
      * `get_optimizer_param_groups_weight_decay()`
    * Regroup the parameters to create the minimal amount of groups.
    * Print all parameter groups when the optimizer is created.
        Param group 0: {amsgrad: False, betas: (0.9, 0.999), eps: 1e-08, lr: 10.0, params: 1, weight_decay: 1.0}
        Param group 1: {amsgrad: False, betas: (0.9, 0.999), eps: 1e-08, lr: 1.0, params: 1, weight_decay: 1.0}
        Param group 2: {amsgrad: False, betas: (0.9, 0.999), eps: 1e-08, lr: 1.0, params: 2, weight_decay: 0.0}
    * Add some unit tests.
    
    Reviewed By: zhanghang1989
    
    Differential Revision: D31287783
    
    fbshipit-source-id: e87df0ae0e67343bb2130db945d8faced44d7411
    2dc3bc02
test_optimizer.py 8.04 KB