dlrm_s_pytorch.py (1 line): - line 1332: # TODO check this lr setup