-
Notifications
You must be signed in to change notification settings - Fork 382
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
oom problem during train #191
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
my gpu is gv100 with 12gb vram
and my config is
parent: projects/neuralangelo/configs/base.yaml
data:
readjust:
center:
- 0.0
- 0.0
- 0.0
scale: 1.0
root: /home/xiangyue/joonwon/neuralangelo/data/owl/
train:
image_size:
- 800
- 600
type: projects.neuralangelo.data
val:
image_size:
- 401
- 300
model:
appear_embed:
enabled: false
object:
sdf:
encoding:
dict_size: 19
dim: 4
coarse2fine:
init_active_level: 4
mlp:
inside_out: false
but occur oom problem
how can i solve it??
I tried reducing the train image size and also tried dict-size to 17.
The text was updated successfully, but these errors were encountered: