--- title: Actor-critic Model keywords: fastai sidebar: home_sidebar summary: "RL Actor-critic model framework." description: "RL Actor-critic model framework." nb_path: "nbs/models/actor_critic.ipynb" ---
embedded_state_size = 64
embedded_action_size = 5
actor_hidden_sizes = (128, 64)
critic_hidden_sizes = (32, 16)
actor = Actor(embedded_state_size=embedded_state_size,
action_weight_size=embedded_action_size,
hidden_sizes=actor_hidden_sizes)
critic = Critic(embedded_state_size=embedded_state_size,
embedded_action_size=embedded_action_size,
hidden_sizes=critic_hidden_sizes)
torch.manual_seed(0)
actor.forward(torch.rand(2,embedded_state_size))
torch.manual_seed(0)
critic.forward(torch.rand(2,embedded_state_size),torch.rand(2,embedded_action_size))