12wang3 / rrl

The code of NeurIPS 2021 paper "Scalable Rule-Based Representation Learning for Interpretable Classification" and TPAMI paper "Learning Interpretable Rules for Scalable Data Representation and Classification"
MIT License
106 stars 32 forks source link

请教一个关于运行效率的问题。 #10

Closed xhxt2008 closed 1 year ago

xhxt2008 commented 1 year ago

请教一下,我用您的测试数据可以跑得起来,但是执行效率看起来有点低,您是否知道可能是什么问题,如何排查。

image

image

想请教一下您,这样的执行效率是否正常,是否是可能因为环境问题或者配置错误导致的效率低下呢。

12wang3 commented 1 year ago

看描述比较奇怪,tic-tac-toe数据集的话,正常来说400个epoch可能也用不了5min。猜测是torch或者CUDA的版本问题,可以尝试换一个版本。

xhxt2008 commented 1 year ago

研究了一下,这个问题应该是我这边环境的问题,不是代码本身的问题。