• Benjamin Lefaudeux's avatar
    [feat] OSS: Sync all attributes (#67) · 5a268b25
    Benjamin Lefaudeux authored
    Make sure that all attributes (not just LR) are in sync in between the OSS.param_groups and the actual wrapped optimizer. Some frameworks make it possible to alter any attribute on a scheduled basis, which proves useful depending on the optimizer, so the keys need to be generically supported (not just "lr"). Not syncing these attributes is a worst case scenario, since these adjustments are silently not propagated, fixing that. 
    5a268b25
test_oss.py 9.67 KB