Skip to content

GitLab

Commit 4ae61134 authored by wangwei990215's avatar wangwei990215
Browse files

Update train.py

parent 3b17f565
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
paddleseg/core/train.py
View file @ 4ae61134
...@@ -241,7 +241,7 @@ def train(model, ...@@ -241,7 +241,7 @@ def train(model,
if (iter) % log_iters == 0: if (iter) % log_iters == 0:
avg_loss /= log_iters avg_loss /= log_iters
avg_loss_list = [l[0] / log_iters for l in avg_loss_list] avg_loss_list = [l / log_iters for l in avg_loss_list]
remain_iters = iters - iter remain_iters = iters - iter
avg_train_batch_cost = batch_cost_averager.get_average() avg_train_batch_cost = batch_cost_averager.get_average()
avg_train_reader_cost = reader_cost_averager.get_average() avg_train_reader_cost = reader_cost_averager.get_average()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment