Searched refs:num_train_batches_ (Results 1 – 2 of 2) sorted by relevance
| /dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/include/ |
| H A D | Engine.hpp | 95 int batch_size_, num_train_batches_, num_test_batches_, num_test_views_; member in MLEngine
166 int get_num_train_batches() { return num_train_batches_; } in get_num_train_batches()
178 void set_num_train_batches(int ntrainb) {num_train_batches_ = ntrainb; } in set_num_train_batches()
|
| /dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/src/ |
| H A D | Engine.cpp | 588 printf("TRAIN mode, epoch %d, training batches %d\n", current_epoch_, num_train_batches_); in run()
596 for(; current_batch_<num_train_batches_; current_batch_++) in run()
628 if(current_batch_ == num_train_batches_-1) in run()
684 if ( num_train_batches_ > 1 ) { in run()
687 …intf("%s; Average Training time = %f seconds", hostname, runtime/((double)(num_train_batches_-2))); in run()
689 …erage Training throughput = %f images/s\n", ((double)(batch_size_*(num_train_batches_-2)))/runtime… in run()
|