|
ConvNet
1.0
A GPU-based C++ implementation of Convolutional Neural Nets
|
Makes data accessible to the model. More...
#include <datahandler.h>
Public Member Functions | |
| DataHandler (const config::DatasetConfig &config) | |
| void | GetBatch (vector< Layer * > &data_layers) |
| int | GetBatchSize () const |
| int | GetDataSetSize () const |
| void | Seek (int row) |
| void | Preprocess (Matrix &input, Matrix &output) |
| void | Sync () |
Protected Member Functions | |
| void | SetupShuffler () |
| void | ShuffleIndices () |
| void | LoadChunk (DataIterator &it, Matrix &mat) |
| void | LoadChunk (DataIterator &it, Matrix &mat, vector< int > &random_rows) |
| void | PipelinedDiskAccess () |
| void | DiskAccess () |
| void | StartPreload () |
| void | WaitForPreload () |
Protected Attributes | |
| default_random_engine | generator_ |
| uniform_int_distribution< int > * | distribution_ |
| map< string, DataIterator * > | data_it_ |
| map< string, Matrix > | data_ |
| vector< string > | layer_names_ |
| thread * | preload_thread_ |
| Matrix | rand_perm_indices_ |
| int | batch_size_ |
| int | chunk_size_ |
| int | max_reuse_count_ |
| int | reuse_counter_ |
| int | random_access_chunk_size_ |
| int | dataset_size_ |
| int | start_ |
| bool | restart_ |
| bool | nothing_on_gpu_ |
| bool | fits_on_gpu_ |
| const bool | pipeline_loads_ |
| const bool | randomize_cpu_ |
| const bool | randomize_gpu_ |
Makes data accessible to the model.
Provides a GetBatch() method that is used by the model to fetch mini-batches. Handles multiple streams of data.
1.8.7