Recommendation pytorch failed

mlcommons / training

Reference implementations of MLPerf™ training benchmarks

https://mlcommons.org/en/groups/training

Apache License 2.0

1.59k stars 553 forks source link

Recommendation pytorch failed #412

Closed LiuXiaoxuanPKU closed 1 year ago

LiuXiaoxuanPKU commented 3 years ago

when running ./run_and_time.sh, I got the following error

Traceback (most recent call last): File "ncf.py", line 421, in main() File "ncf.py", line 231, in main for a, b in zip(test_items, test_pos)]) File "ncf.py", line 231, in for a, b in zip(test_items, test_pos)]) RuntimeError: _th_sort not supported on CPUType for Bool

It fails in L231 of ncf.py:

sorted_items, indices = zip(*[(a != b).sort()
                              for a, b in zip(test_items, test_pos)])

johntran-nv commented 1 year ago

NCF is deprecated, so closing.