mirror of
https://github.com/wassname/ray.git
synced 2026-07-11 20:24:58 +08:00
Amog Kamsetty
14 lines
377 B
Python
14 lines
377 B
Python
from ray.ray_constants import env_integer
|
|
|
|
USE_FP16 = "__use_fp16__"
|
|
NUM_STEPS = "__num_steps__"
|
|
SCHEDULER_STEP = "scheduler_step"
|
|
SCHEDULER_STEP_BATCH = "batch"
|
|
SCHEDULER_STEP_EPOCH = "epoch"
|
|
SCHEDULER_STEP_MANUAL = "manual"
|
|
NCCL_TIMEOUT_S = env_integer("NCCL_TIMEOUT_S", 1800)
|
|
|
|
VALID_SCHEDULER_STEP = {
|
|
SCHEDULER_STEP_BATCH, SCHEDULER_STEP_EPOCH, SCHEDULER_STEP_MANUAL
|
|
}
|