Update train.py

add sorted
amandazw · Jun 26, 2019 · a7909d5 · a7909d5
1 parent 6ce6bed
commit a7909d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/train.py b/tools/train.py
@@ -227,7 +227,7 @@ def is_valid_number(x):
         batch_info = {}
         batch_info['batch_time'] = average_reduce(batch_time)
         batch_info['data_time'] = average_reduce(data_time)
-        for k, v in outputs.items():
+        for k, v in sorted(outputs.items()):
             batch_info[k] = average_reduce(v.data.item())
 
         average_meter.update(**batch_info)