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

[core] Move log_to_driver back to public (#10422)

Browse Source
This commit is contained in:
Richard Liaw
2020-08-29 16:35:14 -07:00
committed by GitHub
parent 9a31166050
commit cb438be146
7 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/torch/examples/cifar_pytorch_example.py
+1 -1
View File
@@ -102,7 +102,7 @@ if __name__ == "__main__":
args, _ = parser.parse_known_args()
num_cpus = 4 if args.smoke_test else None
ray.init(address=args.address, num_cpus=num_cpus, _log_to_driver=True)
ray.init(address=args.address, num_cpus=num_cpus, log_to_driver=True)
trainer1 = TorchTrainer(
model_creator=ResNet18,
python/ray/util/sgd/torch/examples/cifar_pytorch_pbt.py
+1 -1
View File
@@ -98,7 +98,7 @@ if __name__ == "__main__":
"--tune", action="store_true", default=False, help="Tune training")
args, _ = parser.parse_known_args()
ray.init(address=args.address, _log_to_driver=True)
ray.init(address=args.address, log_to_driver=True)
TorchTrainable = TorchTrainer.as_trainable(
model_creator=ResNet18,