-
Notifications
You must be signed in to change notification settings - Fork 103
Open
Description
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
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels