Thank you for your response.
Sorry I am a beginer and I have some additional questions:


What is Dataset in the first line? Should I write the train_set instead? or should I define a new dataset?


What do you mean “using datasets.MNIST” in the def__init__ function? According to my code, I wrote  self.images , self.labels = Data_tr but I got an error.