diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index 9b07354..60769ee 100644 --- a/pokemonred_puffer/environment.py +++ b/pokemonred_puffer/environment.py @@ -949,7 +949,6 @@ def get_game_state_reward(self, print_stats=False): "stats_menu": self.seen_stats_menu * 0.1, "bag_menu": self.seen_bag_menu * 0.1, "cancel_bag_menu": self.seen_cancel_bag_menu * 0.1, - "blackout_check": self.blackout_check * 0.0001, } return state_scores diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py index f12df8f..b901263 100644 --- a/pokemonred_puffer/train.py +++ b/pokemonred_puffer/train.py @@ -227,11 +227,11 @@ def train(args, env_module, make_env): "coords": 0.9995, "map_ids": 0.995, "explore": 0.9995, - "start_menu": 0.95, - "pokemon_menu": 0.95, - "stats_menu": 0.99, - "bag_menu": 0.95, - "cancel_bag_menu": 0.99, + "start_menu": 0.98, + "pokemon_menu": 0.98, + "stats_menu": 0.98, + "bag_menu": 0.98, + "cancel_bag_menu": 0.98, }, "forgetting_frequency": 10, "perfect_ivs": parsed_args.perfect_ivs,