github
/
minimind-v
mirror of https://github.com/jingyaogong/minimind-v.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[fix] add gc

pull/107/head
jingyaogong 2025-12-27 01:09:33 +08:00
parent b7cb98c136
commit 32cf4c5c01
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trainer/trainer_utils.py
View File

@ -4,6 +4,7 @@
import os import os
import random import random
import math import math
import gc
import numpy as np import numpy as np
import torch import torch
import torch.distributed as dist import torch.distributed as dist