ray/rllib/models/tf/__init__.py

15 lines
489 B
Python
Raw Normal View History

# TODO(sven): Add once ModelV1 is deprecated and we no longer cause circular
# dependencies b/c of that.
# from ray.rllib.models.tf.tf_modelv2 import TFModelV2
# from ray.rllib.models.tf.fcnet_v2 import FullyConnectedNetwork
# from ray.rllib.models.tf.recurrent_tf_modelv2 import \
# RecurrentTFModelV2
# from ray.rllib.models.tf.visionnet_v2 import VisionNetwork
# __all__ = [
# "FullyConnectedNetwork",
# "RecurrentTFModelV2",
# "TFModelV2",
# "VisionNetwork",
# ]