-
Notifications
You must be signed in to change notification settings - Fork 2
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
A Error in training #2
Comments
Clip point locations according to [issue-2](#2)
This happened because the point locations may be out of range. I add the
|
Thanks for your thoughtful reply, the problem has been solved. |
I'm afraid the answer is no. However, the batch size is 16, and the memory requirement is about 21G, so one 3090-GPU is enough to train this model. You can reduce the batch size and adjust the learning rate accordingly for training with limited resources. Another way to reduce computation costs is using FP16, which may decrease the accuracy a little but save lots of memory. |
Thanks for your thoughtful reply. |
Thank you for your sharing. May I ask that after data preprocessing according to the code you provided, the training program reported an error:
File "/home/counting/SPDCN-CAC-main/datasets/dataset.py", line 56, in getSample
dotmap[0, points[:, 1], points[:, 0]] = 1
IndexError: index 384 is out of bounds for axis 1 with size 384
The text was updated successfully, but these errors were encountered: