-
Notifications
You must be signed in to change notification settings - Fork 36
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Train tgnet_fps and find the error #32
Comments
My base_name_train_fold.txt belike |
I modify the code ,change the parameter of shuffle from true to false,solve this issue。 |
I solve the problem ,txt file should be like |
What's your video card? How long did it take to train tgnet_fps? |
Hello, when I train the tgnet_fps model, I got following error
Traceback (most recent call last):
File "start_train.py", line 51, in
runner(config, model)
File "/workspace/ToothGroupNetwork-main/ToothGroupNetwork-main/runner.py", line 53, in runner
gen_set = [get_generator_set(config["generator"], False)]
File "/workspace/ToothGroupNetwork-main/ToothGroupNetwork-main/runner.py", line 28, in get_generator_set
point_loader = DataLoader(
File "/opt/conda/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 268, in init
sampler = RandomSampler(dataset, generator=generator)
File "/opt/conda/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
How to solve this issue? By the way , I use the challange dataset.
The text was updated successfully, but these errors were encountered: