shivakanthsujit commited on
Commit
987de76
1 Parent(s): 829d2c2

Init Commit

Browse files
Files changed (3) hide show
  1. README.md +17 -1
  2. config.yaml +20 -0
  3. model.pth +1 -1
README.md CHANGED
@@ -1,3 +1,19 @@
1
  ---
2
- {}
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: mbrl-lib
3
+ tags:
4
+ - mbrl-continuous-cartpole
5
+ - deep-reinforcement-learning
6
+ - reinforcement-learning
7
+ - mbrl-lib
8
  ---
9
+
10
+ # **OneDTransitionRewardModel** Agent playing **mbrl-continuous-cartpole**
11
+ This is a trained model of a **OneDTransitionRewardModel** agent playing **mbrl-continuous-cartpole**
12
+ using [MBRL-Lib](https://github.com/facebookresearch/mbrl-lib).
13
+
14
+ ## Usage (with MBRL-Lib)
15
+ TODO: Add your code
16
+ ```python
17
+ from mbrl import ...
18
+ ...
19
+ ```
config.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dynamics_model:
2
+ _target_: mbrl.models.GaussianMLP
3
+ num_layers: 3
4
+ ensemble_size: 5
5
+ device: cpu
6
+ hid_size: 200
7
+ in_size: 5
8
+ out_size: 4
9
+ deterministic: false
10
+ propagation_method: fixed_model
11
+ activation_fn_cfg:
12
+ _target_: torch.nn.LeakyReLU
13
+ negative_slope: 0.01
14
+ algorithm:
15
+ learned_rewards: false
16
+ target_is_delta: true
17
+ normalize: true
18
+ overrides:
19
+ model_batch_size: 32
20
+ validation_ratio: 0.05
model.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8563fc2a32999509e990981003d1a0263e846a3a5f5ab1ea1458f1eed671cb25
3
  size 1667439
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2716043e8b609a70b19653c73e24b8e67e2e58afa9e1079db130f4cc1ccc2185
3
  size 1667439