ray/rllib/models/tf/__init__.py

13 lines
344 B
Python
Raw Normal View History

from ray.rllib.models.tf.tf_modelv2 import TFModelV2
from ray.rllib.models.tf.fcnet import FullyConnectedNetwork
from ray.rllib.models.tf.recurrent_net import \
RecurrentNetwork
from ray.rllib.models.tf.visionnet import VisionNetwork
__all__ = [
"FullyConnectedNetwork",
"RecurrentNetwork",
"TFModelV2",
"VisionNetwork",
]