Skip to content

Commit 6c84c36

Browse files
fix print bug
1 parent e117596 commit 6c84c36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ppsci/solver/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def closure():
209209
solver.train_time_info["reader_cost"].update(reader_cost)
210210
solver.train_time_info["batch_cost"].update(batch_cost)
211211
printer.update_train_loss(solver, loss_dict, total_batch_size)
212-
if iter_id == 1 and iter_id % log_freq == 0:
212+
if iter_id == 1 or iter_id % log_freq == 0:
213213
printer.log_train_info(solver, total_batch_size, epoch_id, iter_id)
214214

215215
batch_tic = time.perf_counter()

0 commit comments

Comments
 (0)