mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
4d2056722a
* Do mixup in custom collate fn if prefetcher enabled, reduces performance impact * Move mixup code to own file * Add arg to disable prefetcher * Fix no cuda transfer when prefetcher off * Random erasing when prefetcher off wasn't changed to match new args, fixed * Default random erasing to off (prob = 0.) for train
7 lines
228 B
Python
7 lines
228 B
Python
from data.constants import *
|
|
from data.config import resolve_data_config
|
|
from data.dataset import Dataset
|
|
from data.transforms import *
|
|
from data.loader import create_loader
|
|
from data.mixup import mixup_target, FastCollateMixup
|