Skip to content

train bug #37

@zpz915

Description

@zpz915

python train.py \

--dataset mpiigaze
--snapshot output/snapshots
--gpu 0
--num_epochs 50
--batch_size 16
--lr 0.00001
--arch ResNet101
--alpha 1
Loading data.
45000 items removed from dataset that have an angle > 0
Traceback (most recent call last):
File "train.py", line 276, in
train_loader_gaze = DataLoader(
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 268, in init
sampler = RandomSampler(dataset, generator=generator)
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/sampler.py", line 102, in init
raise ValueError("num_samples should be a positive integer "
ValueError: num_samples should be a positive integer value, but got num_samples=0

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions