search

gafniguy / 4D-Facial-Avatars

Dynamic Neural Radiance Fields for Monocular 4D Facial Avater Reconstruction
679 stars 67 forks source link

How much RAM required for training? #53

Closed anthnyprschka closed 1 year ago

anthnyprschka commented 1 year ago

I am able to set up environment on Windows. Now that I run training I run into this issue:

Traceback (most recent call last):
File "F:\4D-Facial-Avatars\nerface_code\nerf-pytorch\train_transformed_rays.py", line 602, in <module>
main()
File "F:\4D-Facial-Avatars\nerface_code\nerf-pytorch\train_transformed_rays.py", line 62, in main
images, poses, render_poses, hwf, i_split, expressions, , bboxs = loadflame_data(
File "F:\4D-Facial-Avatars\nerface_code\nerf-pytorch\nerf\load_flame.py", line 87, in load_flame_data
imgs = (np.array(imgs) / 255.0).astype(np.float32)
numpy.core._exceptions._ArrayMemoryError: Unable to allocate 4.03 GiB for an array with shape (5507, 512, 512, 3) and data type uint8
BanjiBear commented 8 months ago

@anthnyprschka Dear sir, I am currently exploring the work also on Windows. I ran into the same issue raised by you above. I would like to humbly ask if you could provide any solutions?

Thank you so much in advance