Skip to content

Commit d36da65

Browse files
committed
add width to pkl files in the logger.
1 parent 0f5aeed commit d36da65

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bridger/go_explore_phase_1.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def explore(
423423
)
424424
if hparams.debug:
425425
object_logger.log(
426-
"start_entries.pkl",
426+
"start_entries-width-{hparams.env_width}.pkl",
427427
OccurrenceLogEntry(batch_idx=iteration, object=start_entries),
428428
)
429429

@@ -454,7 +454,7 @@ def explore(
454454

455455
if hparams.debug:
456456
object_logger.log(
457-
f"state_cache-{hparams.env_width}.pkl",
457+
f"state_cache-width-{hparams.env_width}.pkl",
458458
OccurrenceLogEntry(batch_idx=0, object=state_sampler),
459459
)
460460

@@ -479,7 +479,7 @@ def explore(
479479
)
480480

481481
for success_entry in success_entries:
482-
object_logger.log("success_entry.pkl", success_entry)
482+
object_logger.log("success_entry-width-{hparams.env_width}.pkl", success_entry)
483483

484484
print(
485485
f"==========\nEntry Count: {len(success_entries)}\n * wa-sampled: {hparams.go_explore_wa_sampled}\n * wa-new: {hparams.go_explore_wa_led_to_something_new}\n * wa-visit: {hparams.go_explore_wa_times_visited}\nShortest: {sorted([len(x.trajectory) for x in success_entries ])}"

0 commit comments

Comments
 (0)