Skip to content
This repository has been archived by the owner on Sep 27, 2024. It is now read-only.

I encountered the following problems when running the train.py. What is the reason? I use CASIA datasets to run . #6

Open
qqqsd opened this issue Aug 30, 2021 · 0 comments

Comments

@qqqsd
Copy link

qqqsd commented Aug 30, 2021

File "D:/PycharmProjects/work/face_recognition.pytorch-develop/train.py", line 123, in train
for i, data in enumerate(train_loader,0):
File "C:\User\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 435, in next
data = self._next_data()
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 1085, in _next_data
return self._process_data(data)
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 1111, in _process_data
data.reraise()
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch_utils.py", line 428, in reraise
raise self.exc_type(msg)
AssertionError: Caught AssertionError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\PycharmProjects\work\face_recognition.pytorch-develop\datasets\casia.py", line 58, in getitem
d_size=(200, 200), normalized=True, show=False)
File "D:\PycharmProjects\work\face_recognition.pytorch-develop\utils\face_align.py", line 28, in align
assert len(landmarks) == 10
AssertionError

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant