shared_ptr<Caffe::RNG> prefetch_rng_;
const unsigned int prefetch_rng_seed = caffe_rng_rand();
prefetch_rng_.reset(new Caffe::RNG(prefetch_rng_seed));
LOG(INFO) << "Declare prefetch_rng_";
caffe::rng_t* prefetch_rng =
static_cast<caffe::rng_t*>(prefetch_rng_->generator());
LOG(INFO) << "prefetch_rng generated";
shuffle(lines_.begin(), lines_.end(), prefetch_rng);
shuffle vector
最新推荐文章于 2022-03-09 09:53:30 发布