mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
14 lines
507 B
Python
14 lines
507 B
Python
# TODO(sven): Add once ModelV1 is deprecated and we no longer cause circular
|
|
# dependencies b/c of that.
|
|
# from ray.rllib.models.torch.torch_modelv2 import TorchModelV2
|
|
# from ray.rllib.models.torch.fcnet import FullyConnectedNetwork
|
|
# from ray.rllib.models.torch.recurrent_torch_model import \
|
|
# RecurrentTorchModel
|
|
# from ray.rllib.models.torch.visionnet import VisionNetwork
|
|
|
|
# __all__ = [
|
|
# "FullyConnectedNetwork",
|
|
# "RecurrentTorchModel",
|
|
# "TorchModelV2",
|
|
# "VisionNetwork",
|
|
# ]
|