Reduce batch size ot fix CPU memory, test=develop (#29736)

Unit test reported memory not enough on CPU machines. Reduce batch size again.
revert-31562-mean
Huihuang Zheng 4 years ago committed by GitHub
parent 8bd2879ef7
commit 2e788bd81e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -93,7 +93,7 @@ cfg.ignore_thresh = .7
# 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

Loading…
Cancel
Save