[RLlib] IMPALA PyTorch (#8287)

This PR adds an IMPALA PyTorch implementation.

- adds compilation tests for LSTM and w/o LSTM.
- adds learning test for CartPole.
This commit is contained in:
Sven Mika
2020-05-03 13:44:25 +02:00
committed by GitHub
parent 1228369a87
commit 166bb5d690
17 changed files with 335 additions and 45 deletions
@@ -30,10 +30,7 @@ def do_test_explorations(run,
# Test all frameworks.
for fw in framework_iterator(core_config):
if fw == "torch" and \
run in [impala.ImpalaTrainer, sac.SACTrainer]:
continue
elif fw == "eager" and run in [
if fw == "eager" and run in [
ddpg.DDPGTrainer, sac.SACTrainer, td3.TD3Trainer
]:
continue
+1 -1
View File
@@ -47,7 +47,7 @@ def framework_iterator(config=None,
logger.warning(
"framework_iterator skipping torch (not installed)!")
continue
elif not tf:
if fw != "torch" and not tf:
logger.warning("framework_iterator skipping {} (tf not "
"installed)!".format(fw))
continue