Skip to content

Commit 7da06fe

Browse files
committed
Merge branch 'main' into extend-interpret
2 parents 2ceef4b + 283e84b commit 7da06fe

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/reward_preprocessing/interpret.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def _determine_features_are_actions(nmf: LayerNMF, layer_name: str) -> bool:
6262
human-understandable action name instead of the feature index."""
6363
# This is the heuristic for determining whether features are actions:
6464
# - If there is no dim reduction
65-
# - If the name of the layer suggest that we are analysing the final layer of a
65+
# - If the name of the layer suggests that we are analysing the final layer of a
6666
# reward net
6767
# - If the number of features is 15 since that is the number of actions in all
6868
# procgen games

src/reward_preprocessing/scripts/train_gan.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def train_gan(
8787
)
8888
# print out summary
8989
gan.summary()
90+
print("Folder:", gan.folder)
9091
# fit gan
9192
steps = {"Adversary": adv_steps}
9293
gan.fit(

0 commit comments

Comments
 (0)