해결된 질문
24.07.23 21:20 작성
·
54
1
vit.ipynb 파일에서, 코드 !python vit.py --pretrained 1 --drop_rate 0.1 --weight_decay 2e-4 , 이것을 실행하였더니,Traceback (most recent call last): File "/content/gdrive/MyDrive/인프런/VisionTransformer/VisionTransformer/VisionTransformer/vit.py", line 38, in <module> image_patches, _ = iter(trainloader).next() AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next' , 이런 에러가 발생하였는데, 어떻게 해결할 수 있을까요?
답변 1
0
2024. 07. 23. 22:21
안녕하세요.
파이토치 버전이 변경 되면서 해당 기능이 사라졌나 보네요.
iter(trainloader).next()를 next(iter(trainloader))로 변경해서 사용하시면 됩니다.
제보 감사드립니다!
딥러닝호형 드림