- 博客(2)
- 收藏
- 关注
原创 CVPR2023 MMANet 代码梳理与分析
prepare_data(args, ckpt_dir) 根据 args.dataset 加载:训练数据会做 resize、随机尺度增强等,label 还会生成多尺度监督,对应label_downsampling_rates = [8, 16, 32]。但是它把整个 args 继续传给了模型构造函数:model = ESANet_KD_Auxi(..., args=args)然后再model.py的21行会有。核心模型是 ESANet_KD_Auxi,定义在 model.py (line 306)。
2026-06-05 15:09:46
361
原创 Pytorch多进程训练,GPU显存未释放问题排查记录
主训练进程虽然已经退出,但它启动的部分 Python 子进程没有被完整终止。这些子进程仍然持有 GPU 设备文件,因此显存无法正常释放。
2026-06-02 19:10:41
248
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅