AttributeError: module 'torchvision.transforms.functional' has no attribute 'affine'

[yaoyz96]

when I run python train.py --config configs/pspnet_pascal_yml, give me this error:

Traceback (most recent call last): File "train.py", line 229, in train(cfg, writer, logger) File "train.py", line 118, in train for (images, labels) in trainloader: File "/data/zyy/usr/local/anaconda3/envs/pytorch-semseg/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in next return self._process_next_batch(batch) File "/data/zyy/usr/local/anaconda3/envs/pytorch-semseg/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch raise batch.exc_type(batch.exc_msg) AttributeError: Traceback (most recent call last): File "/data/zyy/usr/local/anaconda3/envs/pytorch-semseg/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/data/zyy/usr/local/anaconda3/envs/pytorch-semseg/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in samples = collate_fn([dataset[i] for i in batch_indices]) File "/data/zyy/code/pytorch-semseg-master/ptsemseg/loader/pascal_voc_loader.py", line 92, in getitem im, lbl = self.augmentations(im, lbl) File "/data/zyy/code/pytorch-semseg-master/ptsemseg/augmentations/augmentations.py", line 23, in call img, mask = a(img, mask) File "/data/zyy/code/pytorch-semseg-master/ptsemseg/augmentations/augmentations.py", line 207, in call tf.affine( AttributeError: module 'torchvision.transforms.functional' has no attribute 'affine'

AttributeError: module 'torchvision.transforms.functional' has no attribute 'affine' I have tried torch==0.4.1 and ==1.0, torchvision version==0.2.0, but can't fix it. can anyone help me to solve this probelm?