Skip to content

Commit 79308de

Browse files
FinlaySandersstmio
andcommitted
updated drone_race demo weights for new obs size
Co-authored-by: Sam Turner <[email protected]>
1 parent d0861c3 commit 79308de

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

pufferlib/ocean/drone_race/drone_race.c

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
// Standalone C demo for DroneRace environment
2-
// Compile using: ./scripts/build_ocean.sh drone [local|fast]
3-
// Run with: ./drone
2+
// Compile using: ./scripts/build_ocean.sh drone_race [local|fast]
3+
// Run with: ./drone_race
44

55
#include "drone_race.h"
66
#include "puffernet.h"
@@ -124,16 +124,16 @@ int main() {
124124
env->max_moves = 1000;
125125
env->max_rings = 10;
126126

127-
size_t obs_size = 25;
127+
size_t obs_size = 29;
128128
size_t act_size = 4;
129129
env->observations = (float *)calloc(obs_size, sizeof(float));
130130
env->actions = (float *)calloc(act_size, sizeof(float));
131131
env->rewards = (float *)calloc(1, sizeof(float));
132132
env->terminals = (unsigned char *)calloc(1, sizeof(float));
133133

134-
Weights *weights = load_weights("resources/drone/drone_weights.bin", 136073);
134+
Weights *weights = load_weights("resources/drone_race/drone_race_weights.bin", 136585);
135135
int logit_sizes[1] = {4};
136-
LinearContLSTM *net = make_linearcontlstm(weights, 1, 25, logit_sizes, 1);
136+
LinearContLSTM *net = make_linearcontlstm(weights, 1, obs_size, logit_sizes, 1);
137137

138138
if (!env->observations || !env->actions || !env->rewards) {
139139
fprintf(stderr, "ERROR: Failed to allocate memory for demo buffers.\n");
-532 KB
Binary file not shown.
534 KB
Binary file not shown.

0 commit comments

Comments
 (0)