1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-13 22:56:38 -04:00
ray/rllib/models/tf/__init__.py
Sven Mika bf25aee392
[RLlib] Deprecate all Model(v1) usage. ()
Deprecate all Model(v1) usage.
2020-04-29 12:12:59 +02:00

12 lines
361 B
Python

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",
]