Skip to content

Instantly share code, notes, and snippets.

@ericjang

ericjang/nf1_opt.py

Created January 17, 2018 17:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ericjang/ec388b49bca1d97e6ea81782dab4c467 to your computer and use it in GitHub Desktop.
Save ericjang/ec388b49bca1d97e6ea81782dab4c467 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nf1_opt.py
loss = -tf.reduce_mean(dist.log_prob(x_samples))
train_op = tf.train.AdamOptimizer(1e-3).minimize(loss)
sess = tf.InteractiveSession()
sess.run(tf.global_variables_initializer())
NUM_STEPS = int(1e5)
global_step = []
np_losses = []
for i in range(NUM_STEPS):
_, np_loss = sess.run([train_op, loss])
if i % 1000 == 0:
global_step.append(i)
np_losses.append(np_loss)
if i % int(1e4) == 0:
print(i, np_loss)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment