which part of the code is doing Sec4.5 "Neural Non-rigid Tracking for 3D Reconstruction" of the paper?