fix progress bar counts

Author: levimcclenny

tensordiffeq/fit.py

@@ -42,7 +42,7 @@ def fit(obj, tf_iter=0, newton_iter=0, batch_sz=None, newton_eager=True):
         for epoch in t:
             loss_value = train_op_fn(n_batches, obj)
             # Description will be displayed on the left
-            t.set_description('Adam epoch %i' % epoch)
+            t.set_description('Adam epoch %i' % (epoch+1))
             # Postfix will be displayed on the right,
             # formatted automatically based on argument's datatype
             if epoch % 10 == 0:

tensordiffeq/optimizers.py

@@ -272,7 +272,7 @@ def eager_lbfgs(opfunc, x, state, maxIter=100, learningRate=1, do_verbose=True):
                 print('function value changing less than tolX' + str(tf.abs(f - f_old)))
                 break
 
-            t_.set_description('L-BFGS epoch %i' % epoch)
+            t_.set_description('L-BFGS epoch %i' % (epoch+1))
             if do_verbose:
                 if nIter % 10 == 0:
                     t_.set_postfix(loss=f.numpy())