Skip to content

Commit

Permalink
Reduce batch size ot fix CPU memory, test=develop (PaddlePaddle#29736)
Browse files Browse the repository at this point in the history
Unit test reported memory not enough on CPU machines. Reduce batch size again.
  • Loading branch information
zhhsplendid authored Dec 18, 2020
1 parent 8bd2879 commit 2e788bd
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def __setattr__(self, name, value):
# SOLVER options
#
# batch size
cfg.batch_size = 2 if sys.platform == 'darwin' or os.name == 'nt' else 4
cfg.batch_size = 1 if sys.platform == 'darwin' or os.name == 'nt' else 4
# derived learning rate the to get the final learning rate.
cfg.learning_rate = 0.001
# maximum number of iterations
Expand Down

0 comments on commit 2e788bd

Please sign in to comment.