Data sampler wrapper
| 25 | |
| 26 | |
| 27 | class InfiniteSamplerWrapper(data.sampler.Sampler): |
| 28 | """Data sampler wrapper""" |
| 29 | def __init__(self, data_source): |
| 30 | self.num_samples = len(data_source) |
| 31 | |
| 32 | def __iter__(self): |
| 33 | return iter(InfiniteSampler(self.num_samples)) |
| 34 | |
| 35 | def __len__(self): |
| 36 | return 2 ** 31 |
| 37 | |
| 38 | |
| 39 | def copy_G_params(model): |