mirror of
https://github.com/wassname/ray.git
synced 2026-08-19 12:30:27 +08:00
[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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user