Skip to content

Commit b694993

Browse files
authored
Merge pull request #23 from HumanCompatibleAI/print_gan_folder
Print GAN save folder at start of training
2 parents 68d7275 + c58266b commit b694993

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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)