Convergence Error for Different Dataset

[vivek-thomas-geo]

Hello
I was trying to use the code to rune another binary image model
where N=100 ,which implies its a 100100100 voxel dataset
Shape : (100, 100, 100)
All values are binary (0s and 1s)
Value counts: {np.uint8(0): np.int64(232556), np.uint8(1): np.int64(767444)}
these are the specs of the data i used

the the code is not converging and i tried running for morethan 10 hr for such a small data but didnt work
this is the issue

:47: RuntimeWarning: invalid value encountered in sqrt
denom = np.sqrt(np.real(fou_sig[0,0,c,c,c]**2 + fou_sig[1,1,c,c,c]**2 +
:51: RuntimeWarning: invalid value encountered in scalar divide
if nom/denom <1.e-12 and iiter[i]>0: break
i tried adding a small value to denom but it didnt work
i would be grateful if you could help me fix this