Mikayla Gawarecki 383d2d1f6c Add testing and fix issues for weights_only load for LRScheduler (#123775) ...

Fixes https://github.com/pytorch/pytorch/issues/98921

There were two issues detected:
- `MultiStepLR`: issue is described in https://github.com/pytorch/pytorch/issues/98921, this is resolved by allowlisting `collections.Counter`
- `OneCycleLR`: `state_dict['anneal_func']` is either `<function OneCycleLR._annealing_cos at 0x7f364186f5b0>` or
`<function OneCycleLR._annealing_linear at 0x7f39aa483640>` depending on the `anneal_func` kwarg.
   This leads to `WeightsUnpickler error: Unsupported class __builtin__.getattr` from the `weights_only` Unpickler.

  Fixed the above in a BC-compatible manner by adding `OneCyclicLR._anneal_func_type` as a string attribute and removing `OneCyclicLR.anneal_func`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/123775
Approved by: https://github.com/albanD, https://github.com/malfet

2024-04-16 20:29:27 +00:00

..

test_lrscheduler.py

Add testing and fix issues for weights_only load for LRScheduler (#123775)

2024-04-16 20:29:27 +00:00

test_optim.py

Enable UFMT on test/onnx_caffe2, test/optim, test/package and test/profiler (#123901)

2024-04-15 17:46:59 +00:00

test_swa_utils.py

Enable UFMT on test/onnx_caffe2, test/optim, test/package and test/profiler (#123901)

2024-04-15 17:46:59 +00:00