qiwei_ji

Results 3 comments of qiwei_ji

新建了一个分支,在6月初的master下做了点改动,编译后为0.8.0+cu112版本。链接:https://github.com/Oneflow-Inc/oneflow/tree/Lbfgs [code.zip](https://github.com/Oneflow-Inc/oneflow/files/9348207/code.zip) 训练代码及优化算法代码如上附件所示。解压后代码可直接运行pinns_oneflow.py。开头有定义cuda or cpu device。 结果会打印出来并以图片形式展现。可以主要关注第二幅图片,更为直观,图片内容为真实方程解与神经网络拟合函数解的重合程度。

tanh_grad报错。具体信息如下: oneflow._oneflow_internal.exception.Exception: Check failed: (IsClassRegistered(proto().op_type_name())) The gradient function for op tanh_grad is not found. Please check whether it has been implemented and registered correctly. 下边是个报错的小demo。 [test.zip](https://github.com/Oneflow-Inc/oneflow/files/9357621/test.zip)

提交了第一部分。 算法内部有个小问题: lbfgs的torch版本里有个to_dense()方法解决稀疏矩阵等问题的,不过这个if分支暂时是应该没有跑到。 oneflow里没有找到类似算子。所以实现过程中直接把这部分注释掉了,不知道会不会在一些特殊任务上报错?