Skip to content

请问能否帮忙解决一下,使用DEEPFM会存在内存泄漏的问题 #545

@Menghan-Jia

Description

@Menghan-Jia

Describe the bug(问题描述)
A clear and concise description of what the bug is.Better with standalone code to reproduce the issue.
在使用 deepctr 库中的 DeepFM 模型时,调用 predict 方法会出现内存持续增长的问题,导致内存泄漏。该问题在重复预测(如批量处理或 API 服务)时尤为明显。

To Reproduce(复现步骤)
Steps to reproduce the behavior:

  1. 安装对应版本依赖

  2. model训练,重复调用预测

  3. 观察内存使用:使用 tracemalloc 或 psutil 监控内存,也可以from memory_profiler import profile看每一个语句运行后内存增加状况

  4. See error

Image

Operating environment(运行环境):

  • python version [3.8.16]
  • tensorflow version [2.10.0]
  • deepctr version [0.9.3]

Additional context
Add any other context about the problem here.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions