[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
+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