thanks for your reply @albanD, i solved that problem, now i have an other error, going to post here… i believe you can help.
TypeError: ‘list’ object is not callable
Traceback (most recent call last):
File “/home/akb/pytorch_learning/main.py”, line 141, in 
main()
File “/home/akb/pytorch_learning/main.py”, line 133, in main
train(args,epoch_index,train_loader,model,optimizer,criterion)
File “/home/akb/pytorch_learning/main.py”, line 13, in train
for batch_idx,(data,target) in enumerate(train_loader):
File “/home/akb/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 314, in next
batch = self.collate_fn([self.dataset[i] for i in indices])
File “/home/akb/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 314, in 
batch = self.collate_fn([self.dataset[i] for i in indices])
File “/home/akb/anaconda3/lib/python3.6/site-packages/torchvision-0.2.1-py3.6.egg/torchvision/datasets/mnist.py”, line 77, in getitem
File “/home/akb/anaconda3/lib/python3.6/site-packages/torchvision-0.2.1-py3.6.egg/torchvision/transforms/transforms.py”, line 49, in call
TypeError: ‘list’ object is not callable
Thanks in advance