Introduce Stratified Kcrossfold validation.

[akshay772]

I am new to torch. Can you tell me how to introduce k cross fold validation with the training part in train_attribute_model(). I think some steps will be added after the data is loaded ie, after AttributeDataset class object is returned.