Files
ray/rllib/models/tf/__init__.py
T
Sven Mika bf25aee392 [RLlib] Deprecate all Model(v1) usage. (#8146)
Deprecate all Model(v1) usage.
2020-04-29 12:12:59 +02:00

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