[HELP] Change Sampler and Add Layer

Hi Everyone,
I am trying to add a new fully connected layer and i know I can do that with xtra_fc=[out_size_of_layer]
But I am not able to figure out how to set a different sampler say WeightedRandomSampler from pytorch.
Can anyone help?
Got to submit my project in a next 10 hours, and I am super stuck :stuck_out_tongue: