springtostring / bpr-torch Goto Github PK

python main.py 没法运行啊 大佬求回复

我在window10平台，pycharm运行代码
出现
n_train=335351, n_test=84790, sparsity=0.00092 cuda:0 Train on 335351 samples, 168 steps per epoch epoch 0 15.64s train loss is [0.6923 = 0.6923 + 0.0000] multiprocessing.pool.RemoteTraceback: """ Traceback (most recent call last): File "D:\Users\DBIS\Anaconda3\envs\DGNNRec\lib\multiprocessing\pool.py", line 119, in worker result = (True, func(*args, **kwds)) File "D:\Users\DBIS\Anaconda3\envs\DGNNRec\lib\multiprocessing\pool.py", line 44, in mapstar model.fit(learning_rate=0.001, batch_size=2000, epochs=50, verbose=5, early_stop=False) File "D:\projects\BPR-Torch\model\bpr.py", line 227, in fit result = self.test(batch_size=batch_size) File "D:\projects\BPR-Torch\model\bpr.py", line 276, in test batch_result = pool.map(test_one_user, user_batch_rating_uid) File "D:\Users\DBIS\Anaconda3\envs\DGNNRec\lib\multiprocessing\pool.py", line 266, in map return self._map_async(func, iterable, mapstar, chunksize).get() File "D:\Users\DBIS\Anaconda3\envs\DGNNRec\lib\multiprocessing\pool.py", line 644, in get raise self._value AttributeError: 'int' object has no attribute 'n_items'
错误，请问如何解决？