search

BobMcDear / neural-network-cuda

Neural network from scratch in CUDA/C++
GNU General Public License v3.0
66 stars 12 forks source link

Using less data #3

Closed yy1401 closed 2 years ago

yy1401 commented 2 years ago

Hello,

I was wondering if it would be possible to use the code with fewer data samples? For example, currently there are 100,000 rows, but would it be possible to train only on 10,000 data points?

Thank you!

BobMcDear commented 2 years ago

Hello,

Yes, that should be easy. You could simply delete however many data points you do not need from the CSV files and adjust the batch size in the training script accordingly.

Please let me know if this answers your question.

yy1401 commented 2 years ago

That's very helpful, thanks again!