@@ -312,34 +312,29 @@ def report_iteration(self, x):
312
312
self ._k_counter += 1
313
313
314
314
if self ._k_counter == 1 :
315
- print (
316
- "\n k iteration Energy dE Ngvec ev Ngm ev* ||g||"
317
- )
318
- print (
319
- "--------------------------------------------------------------------------------------------------"
320
- )
315
+ header = "\n k iteration Energy dE"
316
+ if self ._use_analytic_grad :
317
+ header += " Ngvec ev Ngm ev* ||g||"
318
+ header += "\n ------------------------------------------------------"
319
+ if self ._use_analytic_grad :
320
+ header += "--------------------------------------------"
321
+ print (header )
321
322
if self ._print_summary_file :
322
- f = open ("summary.dat" , "w+" , buffering = 1 )
323
- f .write (
324
- "\n # k iteration Energy dE Ngvec ev Ngm ev* ||g||"
325
- )
326
- f .write (
327
- "\n #--------------------------------------------------------------------------------------------------"
328
- )
329
- f .close ()
323
+ header .replace ("\n " , "\n # " ).replace ("\n -" , "\n #--" )
324
+ with open ("summary.dat" , "w+" , buffering = 1 ) as f :
325
+ f .write (header )
330
326
331
327
# else:
332
328
dE = self ._curr_energy - self ._prev_energy
333
- print (
334
- f" { self ._k_counter :7} { self ._curr_energy :+12.10f} { dE :+12.10f} { self ._res_vec_evals :4} { self ._res_m_evals :6} { self ._curr_grad_norm :+12.10f} "
335
- )
329
+ update = f" { self ._k_counter :7} { self ._curr_energy :+12.10f} { dE :+12.10f} "
330
+ if self ._use_analytic_grad :
331
+ update += f" { self ._res_vec_evals :4} { self ._res_m_evals :6} { self ._curr_grad_norm :+12.10f} "
332
+ print (update )
336
333
337
334
if self ._print_summary_file :
338
- f = open ("summary.dat" , "a" , buffering = 1 )
339
- f .write (
340
- f"\n { self ._k_counter :7} { self ._curr_energy :+12.12f} { dE :+12.12f} { self ._res_vec_evals :4} { self ._res_m_evals :6} { self ._curr_grad_norm :+12.12f} "
341
- )
342
- f .close ()
335
+ with open ("summary.dat" , "a" , buffering = 1 ) as f :
336
+ update = "\n " + update
337
+ f .write (header )
343
338
344
339
self ._prev_energy = self ._curr_energy
345
340
0 commit comments