ComeçarComece gratuitamente

A taxa de probabilidade cortada

Agora você implementará a razão de probabilidade recortada, um componente essencial da função objetiva do PPO.

Para referência, a razão de probabilidade é definida como: $$\frac{\pi_\theta(a_t|s_t)}{\pi_{\theta_{old}}(a_t|s_t)}$$

E a razão de probabilidade cortada é: \(\mathrm{clip}(r_t(\theta), 1-\varepsilon, 1+\varepsilon)\).

Este exercício faz parte do curso

Aprendizado por reforço profundo em Python

Ver Curso

Instruções de exercício

  • Obtenha a probabilidade de ação prob em action_log_prob e prob_old em action_log_prob_old.
  • Remova o antigo prob de registro de ações do gráfico de computação do gradiente da tocha.
  • Calcule a razão de probabilidade.
  • Corte o objetivo substituto.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

log_prob = torch.tensor(.5).log()
log_prob_old = torch.tensor(.4).log()

def calculate_ratios(action_log_prob, action_log_prob_old, epsilon):
    # Obtain prob and prob_old
    prob = ____
    prob_old = ____
    # Detach the old action log prob
    prob_old_detached = ____.____()
    # Calculate the probability ratio
    ratio = ____ / ____
    # Apply clipping
    clipped_ratio = torch.____(ratio, ____, ____)
    print(f"+{'-'*29}+\n|         Ratio: {str(ratio)} |\n| Clipped ratio: {str(clipped_ratio)} |\n+{'-'*29}+\n")
    return (ratio, clipped_ratio)

_ = calculate_ratios(log_prob, log_prob_old, epsilon=.2)
Editar e executar código