Hi, I'm trying to run the Geometric Bias part of your code, but I met the problem that function 'find_surface_points' can not find surface points in a batch after about 220 iters, bias loss become NaN. Can you help me to figure out this problem or do you meet the same situation？
Hi, I'm trying to run the Geometric Bias part of your code, but I met the problem that function 'find_surface_points' can not find surface points in a batch after about 220 iters, bias loss become NaN. Can you help me to figure out this problem or do you meet the same situation？