Skip to content

Commit

Permalink
fix demo/prune log (PaddlePaddle#599)
Browse files Browse the repository at this point in the history
* fix prune doc

* fix prune demo batchsize

* fix lr (PaddlePaddle#593)

* fix lr schedule in prune demo (PaddlePaddle#595)

* fix prune demo batchsize

* fix lr shcedule in prune demo
;

Co-authored-by: wanghaoshuang <wanghaoshuang@baidu.com>

* remove softmax from demo/models (PaddlePaddle#596)

* fix prune demo batchsize

* fix lr shcedule in prune demo
;

* remove softmax from demo/models

Co-authored-by: wanghaoshuang <wanghaoshuang@baidu.com>

* fix prune demo log

Co-authored-by: wanghaoshuang <wanghaoshuang@baidu.com>
  • Loading branch information
yukavio and wanghaoshuang authored Jan 13, 2021
1 parent 39f7ed3 commit c83eedc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion demo/prune/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ def train(epoch, program):

for i in range(args.num_epochs):
train(i, train_program)
if i % args.test_period == 0:
if (i + 1) % args.test_period == 0:
test(i, pruned_val_program)
save_model(exe, pruned_val_program,
os.path.join(args.model_path, str(i)))
Expand Down

0 comments on commit c83eedc

Please sign in to comment.