jkvt2 / latent-pose-descriptors Goto Github PK

python train.py

Traceback (most recent call last):
  File "/home/nitin/Latent-Pose-Descriptors/train.py", line 26, in <module>
    dataLoader = DataLoader(dataset, batch_size=batch_size, shuffle=True,
  File "/home/nitin/anaconda3/envs/latent_env/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 266, in __init__
    sampler = RandomSampler(dataset, generator=generator)  # type: ignore
  File "/home/nitin/anaconda3/envs/latent_env/lib/python3.9/site-packages/torch/utils/data/sampler.py", line 102, in __init__
    if not isinstance(self.num_samples, int) or self.num_samples <= 0:
  File "/home/nitin/anaconda3/envs/latent_env/lib/python3.9/site-packages/torch/utils/data/sampler.py", line 110, in num_samples
    return len(self.data_source)
  File "/home/nitin/Latent-Pose-Descriptors/dataset/dataset_class.py", line 84, in __len__
    return self.vid_num[-1]
IndexError: index -1 is out of bounds for axis 0 with size 0

Traceback (most recent call last):
  File "train.py", line 208, in <module>
    optimizer.step()
  File "/home/ken/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 15, in decorate_context
    return func(*args, **kwargs)
  File "/home/ken/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/optim/adam.py", line 99, in step
    exp_avg.mul_(beta1).add_(grad, alpha=1 - beta1)
RuntimeError: The size of tensor a (12) must match the size of tensor b (6) at non-singleton dimension 0