final.ckpt

Results 2 issues of final.ckpt

Tensorflow.InvalidArgumentError: In[0] and In[1] ndims must be == 2: 3 it works well in python or ndim==2

### Description 我想在 oneflow 中实现 [谷歌的Lion优化器](https://arxiv.org/abs/2302.06675) 和 [苏剑林博客中介绍的Tiger优化器](https://kexue.fm/archives/9512)。 当我使用 nn.Graph 时,报错说找不到 `_generate_conf_for_graph()` 函数。 翻了一下源码,发现好像所有内置 optimizer 都是调用了 C api 。 请问 oneflow 的静态图模式是不支持自定义 optimizer 吗?如果支持,我应该怎么实现?如果有计划实现,请问大概什么时候能用? 谢谢。 ### Alternatives _No response_