mirror of
https://github.com/wassname/ray.git
synced 2026-07-16 11:21:10 +08:00
3a134c7224
* Start renaming pytorch to torch * Rename PyTorchTrainer to TorchTrainer * Rename PyTorch runners to Torch runners * Finish renaming API * Rename to torch in tests * Finish renaming docs + tests * Run format + fix DeprecationWarning * fix * move tests up * rename Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
8 lines
219 B
Python
8 lines
219 B
Python
USE_FP16 = "__use_fp16__"
|
|
BATCH_COUNT = "batch_count"
|
|
SCHEDULER_STEP = "scheduler_step"
|
|
SCHEDULER_STEP_BATCH = "batch"
|
|
SCHEDULER_STEP_EPOCH = "epoch"
|
|
|
|
VALID_SCHEDULER_STEP = {SCHEDULER_STEP_BATCH, SCHEDULER_STEP_EPOCH}
|