wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-16 11:27:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Partially Use f string (#10218)

Browse Source 
* flynt. trial 1.

* Trial 1.

* Addressed code review.
This commit is contained in:
SangBin Cho
2020-08-20 18:21:16 -07:00
committed by GitHub
parent 07cd815e5a
commit 92664249e8
41 changed files with 195 additions and 238 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/torch/training_operator.py
+7 -13
View File
@@ -72,23 +72,18 @@ class TrainingOperator:
self._models = models # List of models
assert isinstance(
models,
Iterable), ("Components need to be iterable. Got: {}".format(
type(models)))
Iterable), (f"Components need to be iterable. Got: {type(models)}")
self._optimizers = optimizers # List of optimizers
assert isinstance(
optimizers,
Iterable), ("Components need to be iterable. Got: {}".format(
type(optimizers)))
assert isinstance(optimizers, Iterable), (
f"Components need to be iterable. Got: {type(optimizers)}")
self._train_loader = train_loader
self._validation_loader = validation_loader
self._world_rank = world_rank
self._criterion = criterion
self._schedulers = schedulers
if schedulers:
assert isinstance(
schedulers,
Iterable), ("Components need to be iterable. Got: {}".format(
type(schedulers)))
assert isinstance(schedulers, Iterable), (
f"Components need to be iterable. Got: {type(schedulers)}")
self._config = config
self._use_fp16 = use_fp16
self._device_ids = device_ids
@@ -165,10 +160,9 @@ class TrainingOperator:
desc = ""
if info is not None and "epoch_idx" in info:
if "num_epochs" in info:
desc = "{}/{}e".format(info["epoch_idx"] + 1,
info["num_epochs"])
desc = f"{info['epoch_idx'] + 1}/{info['num_epochs']}e"
else:
desc = "{}e".format(info["epoch_idx"] + 1)
desc = f"{info['epoch_idx'] + 1}e"
_progress_bar = tqdm(
total=info[NUM_STEPS] or len(self.train_loader),
desc=desc,