ajbrock / BigGAN-PyTorch

The author's officially unofficial PyTorch BigGAN implementation.
MIT License
2.84k stars 470 forks source link

Why train_transform wasn't applied in utils.get_data_loaders() #58

Open colinshenc opened 4 years ago

colinshenc commented 4 years ago

Hi, I am building my own model on top of this code and am trying to speed up i/o speed so I am working on using hdf5 file for my custom datasets input. But in get_data_loaders function in utils, it seems like 'train_transform' was not applied and I could not find appropriate image preprocessing/transformation applied anywhere upstream or downstream in the data-feeding pipeline if using hdf5. Would you care to explain why 'train_transform' was not applied if using dataset format hdf5? Or if it was applied, would you point out exactly where it was applied? Thanks very much!