Commit Graph

4 Commits (82ce70eec33eb0be96d6e55fcd46d9e1a1c82f77)

Author SHA1 Message Date
Ross Wightman 02ae11e526 Leaving repeat aug sampler indices as tensor thrashes worker shared process memory
4 years ago
Hyeongchan Kim a0b2657497
Use `torch.repeat_interleave()` to generate repeated indices faster (#1058)
4 years ago
Ross Wightman f262137ff2 Add RepeatAugSampler as per DeiT RASampler impl, showing promise for current (distributed) training experiments.
4 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
6 years ago