pytorch项目迁移，训练过程中被killed

| name       | about	                             | labels   |
| ---------- | ------------------------------------- | -------- |
| Bug Report | Use this template for reporting a bug | kind/bug |

### Describe the current behavior / 问题描述 (Mandatory / 必填)
 将pytorch项目迁移到mindspore，在训练过程中每次到这个位置都会被killed，grad_fn内部报的killed。初步认为是爆显存或内存了。凭借网上资料难以自己解决。
![输入图片说明](https://foruda.gitee.com/images/1705894069587662935/b09e2716_13862544.png "屏幕截图")
 由于原项目使用的c++扩展，于是在调用这些扩展函数的地方，我将传入的tensor转换成torch.tensor。
### Environment / 环境信息 (Mandatory / 必填)
- **Hardware Environment(`Ascend`/`GPU`/`CPU`)  / 硬件环境**: 
> Please delete the backend not involved / 请删除不涉及的后端:
> /device 启智平台智算GPU/CPU

- **Software Environment / 软件环境 (Mandatory / 必填)**:
-- MindSpore version (e.g., 1.7.0.Bxxx) :mindspore2.2.0cpu版本
-- Python version (e.g., Python 3.7.5) :python3.7
-- OS platform and distribution (e.g., Linux Ubuntu 16.04):Ubuntu18.04
-- GCC/Compiler version (if compiled from source): 
cuda10.2
pytorch1.9.1
- **Excute Mode / 执行模式 (Mandatory / 必填)(`PyNative`/`Graph`)**: 
> Please delete the mode not involved / 请删除不涉及的模式:
> /mode pynative

### Related testcase / 关联用例 (Mandatory / 必填)
 
### Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
1. 仓库：https://openi.pcl.ac.cn/NoColorZheng/okgr_last.git
2.  不建议运行setup.py，相应的.so文件已经存储在指定位置
3. 直接运行train.py，数据等文件已经放在指定位置

### Describe the expected behavior / 预期结果 (Mandatory / 必填)
 如何释放内存，不让程序被kill
### Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
 
### Special notes for this issue/备注 (Optional / 选填)

GVP MindSpore/mindspore

内容风险标识

评论 (16)

GVPMindSpore/mindspore

内容风险标识