Skip to content
Snippets Groups Projects
Unverified Commit 1e5056ea authored by Ma Zerun's avatar Ma Zerun Committed by GitHub
Browse files

Stop using generic `Sampler` since it's not supported in early version (#56)

PyTorch.
parent 5170676a
No related branches found
No related tags found
No related merge requests found
......@@ -16,7 +16,7 @@ sync_random_seed = MagicMock(return_value=0)
@DATA_SAMPLERS.register_module()
class DefaultSampler(Sampler[int]):
class DefaultSampler(Sampler):
"""The default data sampler for both distributed and non-distributed
environment.
......@@ -109,7 +109,7 @@ class DefaultSampler(Sampler[int]):
@DATA_SAMPLERS.register_module()
class InfiniteSampler(Sampler[int]):
class InfiniteSampler(Sampler):
"""It's designed for iteration-based runner and yields a mini-batch indices
each time.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment