mirror of
https://github.com/wassname/ray.git
synced 2026-07-03 07:44:45 +08:00
[sgd] Replaced class Resources in sgd with use_gpu (#5252)
This commit is contained in:
@@ -10,7 +10,7 @@ from ray.rllib.optimizers import AsyncSamplesOptimizer
|
||||
from ray.rllib.optimizers.aso_tree_aggregator import TreeAggregator
|
||||
from ray.rllib.utils.annotations import override
|
||||
from ray.tune.trainable import Trainable
|
||||
from ray.tune.trial import Resources
|
||||
from ray.tune.resources import Resources
|
||||
|
||||
# yapf: disable
|
||||
# __sphinx_doc_begin__
|
||||
|
||||
@@ -24,7 +24,8 @@ from ray.rllib.utils.memory import ray_get_and_free
|
||||
from ray.rllib.utils import try_import_tf
|
||||
from ray.tune.registry import ENV_CREATOR, register_env, _global_registry
|
||||
from ray.tune.trainable import Trainable
|
||||
from ray.tune.trial import Resources, ExportFormat
|
||||
from ray.tune.trial import ExportFormat
|
||||
from ray.tune.resources import Resources
|
||||
from ray.tune.logger import UnifiedLogger
|
||||
from ray.tune.result import DEFAULT_RESULTS_DIR
|
||||
|
||||
|
||||
@@ -11,7 +11,7 @@ import ray
|
||||
from ray.tests.cluster_utils import Cluster
|
||||
from ray.tune.config_parser import make_parser
|
||||
from ray.tune.result import DEFAULT_RESULTS_DIR
|
||||
from ray.tune.trial import resources_to_json
|
||||
from ray.tune.resources import resources_to_json
|
||||
from ray.tune.tune import _make_scheduler, run_experiments
|
||||
|
||||
EXAMPLE_USAGE = """
|
||||
|
||||
Reference in New Issue
Block a user