mirror of https://github.com/JDAI-CV/fast-reid.git
17 lines
365 B
Python
17 lines
365 B
Python
import unittest
|
|
import sys
|
|
sys.path.append('.')
|
|
from fastreid.data.samplers import TrainingSampler
|
|
|
|
|
|
class SamplerTestCase(unittest.TestCase):
|
|
def test_training_sampler(self):
|
|
sampler = TrainingSampler(5)
|
|
for i in sampler:
|
|
from ipdb import set_trace; set_trace()
|
|
print(i)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|