Polish code

emailweixu-patch-1
Yang Yu 7 years ago
parent f044b23fac
commit 7a935b1701

@ -122,11 +122,9 @@ def main():
PASS_NUM = 100 PASS_NUM = 100
for pass_id in range(PASS_NUM): for pass_id in range(PASS_NUM):
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
need_check = (batch_id + 1) % 10 == 0
# train a mini-batch, fetch nothing # train a mini-batch, fetch nothing
exe.run(feed=feeder.feed(data)) exe.run(feed=feeder.feed(data))
if need_check: if (batch_id + 1) % 10 == 0:
acc_set = [] acc_set = []
avg_loss_set = [] avg_loss_set = []
for test_data in test_reader(): for test_data in test_reader():

Loading…
Cancel
Save