Skip to content

Commit 12b284d

Browse files
committed
fix typo
1 parent 630663b commit 12b284d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

pufferlib/pufferl.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -963,8 +963,8 @@ def train(env_name, args=None, vecenv=None, policy=None, logger=None, early_stop
963963
if early_stop_fn is not None:
964964
should_stop_early = early_stop_fn(logs)
965965
# This is hacky, but need to see if threshold looks reasonable
966-
if 'early_stop_treshold' in logs:
967-
pufferl.logger.log({'environment/early_stop_treshold': logs['early_stop_treshold']}, logs['agent_steps'])
966+
if 'early_stop_threshold' in logs:
967+
pufferl.logger.log({'environment/early_stop_threshold': logs['early_stop_threshold']}, logs['agent_steps'])
968968

969969
if pufferl.global_step > logging_threshold:
970970
all_logs.append(logs)
@@ -1082,7 +1082,7 @@ def stop_if_perf_below(logs):
10821082

10831083
# If metric distribution is percentile, threshold is also logit transformed
10841084
threshold = sweep.get_early_stop_threshold(cost)
1085-
logs['early_stop_treshold'] = max(threshold, -5) # clipping for visualization
1085+
logs['early_stop_threshold'] = max(threshold, -5) # clipping for visualization
10861086

10871087
if sweep.should_stop(max(target_running_mean, metric_val), cost):
10881088
logs['is_loss_nan'] = False

0 commit comments

Comments
 (0)