2020-09-09 17:33:21 +02:00
|
|
|
from ray.rllib.agents.marwil.bc import BCTrainer, BC_DEFAULT_CONFIG
|
2019-04-07 00:36:18 -07:00
|
|
|
from ray.rllib.agents.marwil.marwil import MARWILTrainer, DEFAULT_CONFIG
|
2020-04-07 01:38:50 +02:00
|
|
|
from ray.rllib.agents.marwil.marwil_tf_policy import MARWILTFPolicy
|
|
|
|
from ray.rllib.agents.marwil.marwil_torch_policy import MARWILTorchPolicy
|
2019-01-17 11:00:43 +08:00
|
|
|
|
2020-04-07 01:38:50 +02:00
|
|
|
__all__ = [
|
2020-09-09 17:33:21 +02:00
|
|
|
"BCTrainer",
|
|
|
|
"BC_DEFAULT_CONFIG",
|
2020-04-07 01:38:50 +02:00
|
|
|
"DEFAULT_CONFIG",
|
|
|
|
"MARWILTFPolicy",
|
|
|
|
"MARWILTorchPolicy",
|
|
|
|
"MARWILTrainer",
|
|
|
|
]
|