MCPcopy Create free account
hub / github.com/ReinFlow/ReinFlow / forward

Method forward

model/flow/ft_baselines/fql.py:134–152  ·  view source on GitHub ↗

Generate actions for a batch of observations using the actor with random noise. Args: cond (Dict[str, Tensor]): Dictionary containing the state tensor with key 'state'. Returns: Tensor: Actions of shape (batch, horizon_steps, action_dim).

(self, 
                cond: Dict[str, Tensor], 
                mode:str='onestep')

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 6

loss_criticMethod · 0.45
loss_actorMethod · 0.45
get_logprobsMethod · 0.45
get_actionsMethod · 0.45
get_logprobsMethod · 0.45
get_actionsMethod · 0.45

Calls 1

sampleMethod · 0.45

Tested by

no test coverage detected