Runtime Error in Training

andrewhou1 / GeomConsistentFR

Official Code for Face Relighting with Geometrically Consistent Shadows (CVPR 2022)

https://openaccess.thecvf.com/content/CVPR2022/html/Hou_Face_Relighting_With_Geometrically_Consistent_Shadows_CVPR_2022_paper.html

MIT License

114 stars 17 forks source link

Runtime Error in Training #11

Closed waleedrazakhan92 closed 1 year ago

waleedrazakhan92 commented 1 year ago

Hello I'm trying to train the model. I've cropped and preprocessed the CelebA data and the data loads just fine. However when I start the training in get this error:

geom_training_error

Can you please guide me what might be the case. I've tried printing the "c2_o_depth" in the forward pass and its dimensions are [0, 1, 256, 256] but i can't diagnose whats causing the error.

waleedrazakhan92 commented 1 year ago

Edit(SOLVED): I changed the number of images and Didn't change the placeholders shape to match the number of images, causing the 0 shaped tensors.