"src/vscode:/vscode.git/clone" did not exist on "18ebd57bd80fdd1bb8cbf9af075ba0301705bc6f"
Commit 11cf47ef authored by Jeff Rasley's avatar Jeff Rasley
Browse files

temporarily disable lr unit tests

parent 67176383
......@@ -4,12 +4,17 @@ import argparse
import pytest
import json
import os
import pytest
from common import distributed_test
from simple_model import SimpleModel, SimpleOptimizer, random_dataloader, args_from_dict
from deepspeed.runtime.lr_schedules import LR_RANGE_TEST, ONE_CYCLE, WARMUP_LR, WARMUP_DECAY_LR
from deepspeed.runtime.lr_schedules import WARMUP_MIN_LR, WARMUP_MAX_LR, WARMUP_NUM_STEPS, TOTAL_NUM_STEPS
from deepspeed.runtime.lr_schedules import CYCLE_MIN_LR, CYCLE_MAX_LR
pytest.skip(
"skipping until we can figure out what's causing these to hang inside our CI",
allow_module_level=True)
@pytest.mark.parametrize("scheduler_type,params",
[(WARMUP_LR,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment