Method _save_checkpoint

tasks/AutoClsSST/code/experiment.py:265–287

(
            self,
            target_dir: str,
            epoch: int,
            optimizer: AdamW,
            best_score: float,
            training_stats: Dict[str, List]
    )

        )

    def _save_checkpoint(
            self,
            target_dir: str,
            epoch: int,
            optimizer: AdamW,
            best_score: float,
            training_stats: Dict[str, List]
    ) -> None:
        checkpoint = {
            "epoch": epoch,
            "model": self.model.state_dict(),
            "optimizer": optimizer.state_dict(),
            "best_score": best_score,
            **training_stats
        }
        torch.save(
            checkpoint,
            os.path.join(target_dir, "best.pth.tar")
        )
        logger.info("Model saved successfully")
        
        # 记录最佳模型信息到TensorBoard
        self.writer.add_text('Best Model', f'Epoch: {epoch}, Accuracy: {best_score:.4f}', epoch)

    def _load_checkpoint(
            self,