Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
sign/unsign comp
Browse files Browse the repository at this point in the history
  • Loading branch information
zhreshold committed Apr 23, 2020
1 parent f0eb202 commit 5d0ac0f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/io/dataloader.cc
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ class ThreadedDataLoader : public IIterator<TBlobBatch> {
if (!has_next) return false;
auto samples = sampler_->Value();
auto batch_size = samples.data[0].shape().Size();
auto real_batch_size = batch_size - samples.num_batch_padd;
int real_batch_size = batch_size - samples.num_batch_padd;
const int64_t *idx_ptr = static_cast<int64_t*>(
samples.data[0].data().dptr_);
std::vector<int64_t> idx_ptrs;
Expand Down

0 comments on commit 5d0ac0f

Please sign in to comment.