Skip to content

Commit 075546e

Browse files
committed
fix code format
test=develop
1 parent 6a13e7c commit 075546e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

dygraph/resnet/train.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,20 +143,25 @@ def parse_args():
143143
args = parse_args()
144144
batch_size = args.batch_size
145145

146+
146147
class TimeCostAverage(object):
147148
def __init__(self):
148149
self.reset()
150+
149151
def reset(self):
150152
self.cnt = 0
151153
self.total_time = 0
154+
152155
def record(self, usetime):
153156
self.cnt += 1
154157
self.total_time += usetime
158+
155159
def get_average(self):
156160
if self.cnt == 0:
157161
return 0
158162
return self.total_time / self.cnt
159163

164+
160165
def optimizer_setting(parameter_list=None):
161166

162167
total_images = IMAGENET1000
@@ -493,7 +498,8 @@ def train_resnet():
493498
"[Epoch %d, batch %d] loss %.5f, acc1 %.5f, acc5 %.5f, batch_cost: %.5f s, reader_cost: %.5f s"
494499
% (eop, batch_id, total_loss / total_sample,
495500
total_acc1 / total_sample, total_acc5 / total_sample,
496-
train_batch_cost_avg.get_average(), train_reader_cost_avg.get_average()))
501+
train_batch_cost_avg.get_average(),
502+
train_reader_cost_avg.get_average()))
497503
train_batch_cost_avg.reset()
498504
train_reader_cost_avg.reset()
499505
batch_start = time.time()

0 commit comments

Comments
 (0)