Thanks. But torchvision.datasets has some transform operations. How to work around those is another problem.
I have tried the following:


extract trainloader.dataset.train_data and trainloader.dataset.train_label


reform dataset with my own data (weight tensor).
Data.TensorDataset(train_data, train_label, weight_tensor)


re-define trainloader.


But it won’t keep transform operations. which is causing dimension error during training.