Skip to content

GitLab

"tasks/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "35a3fa8b7cd26b6022839e4d30e5ca5f21c08dd4"
Commit 4b101ece authored by continue's avatar continue Committed by Kai Chen
Browse files

Fix Sampler bug: index out of range when num_extra < len(indice) (#1610) 

* Fix bug: index out of range when num_extra < len(indice)

* Fix bug: index out of range when num_extra < len(indice)

* formatter code
parent 93aa32de
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
mmdet/datasets/loader/sampler.py
View file @ 4b101ece
...@@ -57,7 +57,8 @@ class GroupSampler(Sampler): ...@@ -57,7 +57,8 @@ class GroupSampler(Sampler):
np.random.shuffle(indice) np.random.shuffle(indice)
num_extra = int(np.ceil(size / self.samples_per_gpu) num_extra = int(np.ceil(size / self.samples_per_gpu)
) * self.samples_per_gpu - len(indice) ) * self.samples_per_gpu - len(indice)
indice = np.concatenate([indice, indice[:num_extra]]) indice = np.concatenate(
[indice, np.random.choice(indice, num_extra)])
indices.append(indice) indices.append(indice)
indices = np.concatenate(indices) indices = np.concatenate(indices)
indices = [ indices = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment