In the code: save_images(self, transformed, save_dir) In the documentation: save_images(self, save_dir, transformed)