lr : 5e-1 1e-1 5e-2 1e-2 5e-3 1e-3 5e-4 1e-4 bwd_lr : 5e-1 1e-1 5e-2 1e-2 5e-3 1e-3 5e-4 1e-4 bwd_samples : 1 2 4 8 16 32