WIGRL Dependencies python == 3.6.12 pytorch == 1.10.2 pytorch-lightning==1.5.10 labml == 0.4.161 labml-helpers == 0.4.89 labml-nn == 0.4.133 sklearn Datasets Fdataset Cdataset LRSSL Usage python demo.py