Update chess-mamba-vs-xformer/train_bygame.py

chess-mamba-vs-xformer/train_bygame.py

@@ -527,9 +527,6 @@ while True:
             'config': config,
         }
         checkpoint['grad_norm_history'] = grad_norm_history
-        if init_from == 'anneal':
-            checkpoint['anneal_start_iters'] = anneal_start_iters
-            checkpoint['anneal_decay_iters'] = anneal_decay_iters
         print(f"Max_iters reached. Saving pre-anneal checkpoint to {anneal_checkpoint}")
         torch.save(checkpoint, os.path.join(out_dir, anneal_checkpoint))
         break