-
Notifications
You must be signed in to change notification settings - Fork 10
Description
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 "