Transform arg does not passed into the Dataset construction

MadryLab / robustness

A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.

MIT License

905 stars 181 forks source link

Transform arg does not passed into the Dataset construction #45

Closed LR32768 closed 4 years ago

LR32768 commented 4 years ago

Hi. When I am using your package for getting a RestrictedImageNet dataloader, it seems that the argument of transform_train and transform_test does not affect the output image. Then I read your code and found that the construction does not use the passed-in transform but the default DA.TRANSFORM_TRAIN instead. A small issue, but looking forward to fixing them. THX

andrewilyas commented 4 years ago

This has been fixed on the develop branch, which will be merged in sometime this week.

andrewilyas commented 4 years ago

This should now be fixed on the master branch!