mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
[RLlib] Change the slateq regression learning test with GPU to use torch only (#23168)
This commit is contained in:
parent
7c43c66b6b
commit
6c20e9d898
1 changed files with 5 additions and 0 deletions
|
@ -468,6 +468,11 @@ slateq-interest-evolution-recsim-env:
|
|||
env_config:
|
||||
# Env class specified above takes one `config` arg in its c'tor:
|
||||
config:
|
||||
# GPU support currently does not work for tensorflow. For the learning
|
||||
# test for now, we'll use pytorch, and investigate why tensorflow is not working.
|
||||
# One indication of the issue is that the loss inputs are not flattened from the
|
||||
# original dictionary spaces.
|
||||
framework: torch
|
||||
# Each step, sample `num_candidates` documents using the env-internal
|
||||
# document sampler model (a logic that creates n documents to select
|
||||
# the slate from).
|
||||
|
|
Loading…
Add table
Reference in a new issue