We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents e117596 + 6c84c36 commit b1a2236Copy full SHA for b1a2236
ppsci/solver/train.py
@@ -209,7 +209,7 @@ def closure():
209
solver.train_time_info["reader_cost"].update(reader_cost)
210
solver.train_time_info["batch_cost"].update(batch_cost)
211
printer.update_train_loss(solver, loss_dict, total_batch_size)
212
- if iter_id == 1 and iter_id % log_freq == 0:
+ if iter_id == 1 or iter_id % log_freq == 0:
213
printer.log_train_info(solver, total_batch_size, epoch_id, iter_id)
214
215
batch_tic = time.perf_counter()
0 commit comments