setting both thresholds to N_sma_threshhold

missed 1 of 2 earlier...

Author: lessw2020

ranger.py

@@ -113,7 +113,7 @@ def step(self, closure=None):
                 if group['weight_decay'] != 0:
                     p_data_fp32.add_(-group['weight_decay'] * group['lr'], p_data_fp32)
 
-                if N_sma > 4:
+                if N_sma > N_sma_threshhold:
                     denom = exp_avg_sq.sqrt().add_(group['eps'])
                     p_data_fp32.addcdiv_(-step_size, exp_avg, denom)
                 else: