param depth dim_m gθ sign flip proba gx sign flip proba θ sign flip proba θ keep only sign gx keep only sign gθ keep only sign value 1 2 3 4 5 6 2 4 8 16 32 64 0 0.4 0.45 0 0.4 0.45 0 0.4 0.45 false true false true false true viz loss ⟩