Knockoff Generator for Custom Dataset

[andrey-vinokurov-1991]

Hello,
Thank you for your work. It looks amazing and I'd like to try it on some custom dataset.
Could you please share the script with training generator for ImageNet or for another dataset?