Change dataset member variable once per epoch

[jabbarshah]

@ptrblck Sorry to text you here. Could you please answer to the below problem?
AttributeError: module ‘torchvision.transforms’ has no attribute ‘Scale’