我正试图掌握多处理(在我的Mac上有2个CPU)。我的代码大致如下:
def new_func(i):
Rewards, Times, CummulativeRewards, Weights = Q_learning(model, variants = 2, stages = 3, gamma = 0.975, epsilon = 0.99, batchSize = i, buffer = 3000, randSeed = 42, replay = None)
return CummulativeRewards
from joblib import Parallel, delayed import multiprocessing
BatchSize = [100, 300]
num_cores = multiprocessing.cpu_count(2)
Rewards, Times, CuRew, Weights = Parallel(n_jobs=num_cores)(delayed(new_func)(i) for i in BatchSize)
我得到以下错误:
num_cores = multiprocessing.cpu_count(2)
TypeError: cpu_count() takes 1 positional argument but 2 were given
有什么问题吗?在
cpu计数不接受任何参数。一个位置参数是自实例
multiprocessing
。在https://docs.python.org/3/library/multiprocessing.html#multiprocessing.cpu_count
相关问题 更多 >
编程相关推荐