Function pick

2-cartpole/3-ppo.py:116–119

(state)

        model.load_state_dict(torch.load(SAVE_PATH))

        def pick(state):
            with torch.no_grad():
                logits, _ = model(torch.as_tensor(state))
                return int(torch.distributions.Categorical(logits=logits).sample().item())

        run_test_loop(env, pick)