Skip to content

GitLab

Commit b7ce045c authored by David Andersen's avatar David Andersen Committed by GitHub
Browse files

Merge pull request #2109 from cclauss/patch-5 

adversarial_crypto: ac.eve_optimizer, ac.reset_eve_vars
parents db74c7e6 ecdc7f87
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
adversarial_crypto/train_eval.py
View file @ b7ce045c
...@@ -249,17 +249,17 @@ def train_and_evaluate(): ...@@ -249,17 +249,17 @@ def train_and_evaluate():
if train_until_thresh(s, ac): if train_until_thresh(s, ac):
for _ in xrange(EVE_EXTRA_ROUNDS): for _ in xrange(EVE_EXTRA_ROUNDS):
s.run(eve_optimizer) s.run(ac.eve_optimizer)
print('Loss after eve extra training:') print('Loss after eve extra training:')
doeval(s, ac, EVAL_BATCHES * 2, 0) doeval(s, ac, EVAL_BATCHES * 2, 0)
for _ in xrange(NUMBER_OF_EVE_RESETS): for _ in xrange(NUMBER_OF_EVE_RESETS):
print('Resetting Eve') print('Resetting Eve')
s.run(reset_eve_vars) s.run(ac.reset_eve_vars)
eve_counter = 0 eve_counter = 0
for _ in xrange(RETRAIN_EVE_LOOPS): for _ in xrange(RETRAIN_EVE_LOOPS):
for _ in xrange(RETRAIN_EVE_ITERS): for _ in xrange(RETRAIN_EVE_ITERS):
eve_counter += 1 eve_counter += 1
s.run(eve_optimizer) s.run(ac.eve_optimizer)
doeval(s, ac, EVAL_BATCHES, eve_counter) doeval(s, ac, EVAL_BATCHES, eve_counter)
doeval(s, ac, EVAL_BATCHES, eve_counter) doeval(s, ac, EVAL_BATCHES, eve_counter)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment