SlowGuess/NeurIPS_2023_OCR huggingface dataset:
Input: The parameters  $\theta_{De}^{\pi}$  for Adaptive Action Decoder of actor  $\pi$ , agent  $i$ 's observation  $o_i^t$  and subtask  $k_i^t$  in timestep  $t$ ,
Output: agent  $i$ 's action  $a_{i}^{t}$  in timestep  $t$
1: Compute subtask embedding  $z_{i}^{t}$  with  $z_{i}^{t} = \text{Embedding}(k_{i}^{t})$
2: Compute adaptive subtask semantics  $\phi_i^t$  by Formula (7)
3: Compute action semantics  $\varphi_{i}^{t}$  by Formula (8)
4: Sample action  $a_{i}^{t}$  from the actions' probability distribution  $Pr_{i}^{t}$  computed by Formula (9)
5: Return  $a_{i}^{t}$