hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[air] Minor Doc updates (#25097)

Browse source 
Update a few docs and param names.
This commit is contained in:
xwjiang2010 2022-05-24 17:15:03 -07:00 • committed by GitHub
parent f79b826f31
commit 51dbd99a25
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/ray/ml/train/integrations/sklearn/sklearn_trainer.py
View file

@ -452,7 +452,7 @@ def load_checkpoint(
with checkpoint.as_directory() as checkpoint_path:
estimator_path = os.path.join(checkpoint_path, MODEL_KEY)
with open(estimator_path, "rb") as f:
estimator_path = cpickle.load(f)
estimator = cpickle.load(f)
preprocessor = load_preprocessor_from_dir(checkpoint_path)
return estimator_path, preprocessor
return estimator, preprocessor

2
python/ray/ml/train/integrations/tensorflow/tensorflow_trainer.py
View file

@ -125,7 +125,7 @@ class TensorflowTrainer(DataParallelTrainer):
)
model.fit(tf_dataset)
train.save_checkpoint(
epoch=epoch, model_weights=model.get_weights())
epoch=epoch, model=model.get_weights())
train_dataset = ray.data.from_items(
[{"x": x, "y": x + 1} for x in range(32)])