[fix] add gc

2025-12-27 01:09:33 +08:00 · 2025-12-27 01:09:33 +08:00 · 32cf4c5c01
parent b7cb98c136
commit 32cf4c5c01
1 changed files with 1 additions and 0 deletions
--- 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