num_samples should be a positive ValueError: integer value, but got num_samples=0

boingboing·2024년 6월 24일
(medT) root@d88db720a0e1:/data/5.MedT/Medical-Transformer# sh train.sh
Traceback (most recent call last):
  File "train.py", line 90, in <module>
    dataloader = DataLoader(train_dataset, batch_size=args.batch_size, shuffle=True)
  File "/opt/conda/envs/medT/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 268, in __init__
    sampler = RandomSampler(dataset, generator=generator)
  File "/opt/conda/envs/medT/lib/python3.6/site-packages/torch/utils/data/sampler.py", line 103, in __init__
    "value, but got num_samples={}".format(self.num_samples))
ValueError: num_samples should be a positive integer value, but got num_samples=0

medT train.py 돌리니까 생긴 에러....

원인

num_samples 갯수가 0으로 잡힘.
-> 처음에 데이터셋 경로를 로컬->AI서버로 옮기면서 경로가 틀렸나 싶었는데 상대경로로 이미 잡혀있어서 괜춘..

->알고보니 이미지들 복사 자체가 안됬음. 폴더 형태만 복사됨

해결

->파일을 복사함

0개의 댓글