2020-04-01 07:00:28 +02:00
|
|
|
# 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
|
2020-05-18 17:26:40 +02:00
|
|
|
# from ray.rllib.models.torch.recurrent_net import \
|
|
|
|
|
# RecurrentNetwork
|
2020-04-01 07:00:28 +02:00
|
|
|
# from ray.rllib.models.torch.visionnet import VisionNetwork
|
|
|
|
|
|
|
|
|
|
# __all__ = [
|
|
|
|
|
# "FullyConnectedNetwork",
|
2020-05-18 17:26:40 +02:00
|
|
|
# "RecurrentNetwork",
|
2020-04-01 07:00:28 +02:00
|
|
|
# "TorchModelV2",
|
|
|
|
|
# "VisionNetwork",
|
|
|
|
|
# ]
|
