How to set batch_size value?

josephjaspers / blackcat_tensors

Matrix-Vector Library Designed for Neural Network Construction. cuda (gpu) support, openmp (multithreaded cpu) support, partial support of BLAS, expression template based implementation PTX code generation identical to hand written kernels, and support for auto-differentiation

12 stars 4 forks source link

How to set batch_size value? #57

Open xinsuinizhuan opened 4 years ago

xinsuinizhuan commented 4 years ago

how to set batch_size value? Default is 32, but when i set it to 16，seems the output result is better than 32, but same epochs==5000, train it need more time.

josephjaspers commented 4 years ago

You can use the "set_batch_size" function.