Skip to content

Training HybridNet with Example_Dataset Fails #11

@breezexixixi

Description

@breezexixixi

Why does it appear when I directly train the HybridNet of Example Project ValueError: num_samples should be a positive integer value, but got num_samples=0?

The following are the relevant logs:

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

File "/home/chenweitao/.local/lib/python3.9/site-packages/streamlit/scriptrunner/script_runner.py", line 557, in _run_script
exec(code, module.dict)
File "/home/chenweitao/visual_recognition/jarvis/JARVIS-HybridNet/jarvis/ui/gui/jarvis_gui.py", line 175, in
train_hybridnet_gui(project_box, projectManager.cfg)
File "/home/chenweitao/visual_recognition/jarvis/JARVIS-HybridNet/jarvis/ui/gui/train_gui.py", line 264, in train_hybridnet_gui
train.train_hybridnet(project, num_epochs, weights_keypoint, weights,
File "/home/chenweitao/visual_recognition/jarvis/JARVIS-HybridNet/jarvis/train_interface.py", line 204, in train_hybridnet
train_results = hybridNet.train(training_set, val_set, num_epochs,
File "/home/chenweitao/visual_recognition/jarvis/JARVIS-HybridNet/jarvis/hybridnet/hybridnet.py", line 151, in train
training_generator = DataLoader(
File "/home/chenweitao/.local/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 344, in init
sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type]
File "/home/chenweitao/.local/lib/python3.9/site-packages/torch/utils/data/sampler.py", line 107, in init
raise ValueError("num_samples should be a positive integer "

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