File size: 1,144 Bytes
c6b1667
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
---
tags:
- Pixelcopter-PLE-v0
- reinforce
- reinforcement-learning
- custom-implementation
- deep-rl-class
model-index:
- name: REINFORCE-PixelCopter
  results:
  - task:
      type: reinforcement-learning
      name: reinforcement-learning
    dataset:
      name: Pixelcopter-PLE-v0
      type: Pixelcopter-PLE-v0
    metrics:
    - type: mean_reward
      value: 6.50 +/- 5.99
      name: mean_reward
      verified: false
---

  # **Reinforce** Agent playing **Pixelcopter-PLE-v0**
  This is a trained model of a **Reinforce** agent playing **Pixelcopter-PLE-v0** .
  Portable weights: `model_state_dict.pt`. Define the matching Unit 4 Policy
  (CartPole: two layers; PixelCopter: three layers), instantiate it with
  `state_space`, `action_space`, and `h_size` from `hyperparameters.json`, then
  use `policy.load_state_dict(torch.load(weights_path, map_location="cpu", weights_only=True))`.
  The legacy `model.pt` contains a full Python model and requires its matching class.
  To learn to use this model and train yours check Unit 4 of the Deep Reinforcement Learning Course: https://huggingface.co/deep-rl-course/unit4/introduction