Libtorch regression modeling bug

Thank you! It works!!