def test_batch_idx_with_none(self, init_optimizer): with pytest.warns(DeprecationWarning): scheduler = CyclicLR(init_optimizer) for p_group in init_optimizer.param_groups: assert p_group['initial_lr'] scheduler.batch_step() assert scheduler.last_batch_idx == 1
def test_batch_idx_with_none(self, init_optimizer): scheduler = CyclicLR(init_optimizer) scheduler.batch_step() assert scheduler.last_batch_idx == 0