From 32cf4c5c01337231fd907b92d513de8945594263 Mon Sep 17 00:00:00 2001
From: jingyaogong <gongjy.cs@qq.com>
Date: Sat, 27 Dec 2025 01:09:33 +0800
Subject: [PATCH] [fix] add gc

---
 trainer/trainer_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/trainer/trainer_utils.py b/trainer/trainer_utils.py
index bc813cf..4fc0ceb 100644
--- a/trainer/trainer_utils.py
+++ b/trainer/trainer_utils.py
@@ -4,6 +4,7 @@
 import os
 import random
 import math
+import gc
 import numpy as np
 import torch
 import torch.distributed as dist