我一直在Windows10机器上使用Deep Q Learning。我有0.4.1版本的pytorch和NVIDA显卡。
def select_action(self, state):
probs = F.softmax(self.model(Variable(state, volatile = True))*7)
action = probs.multinomial()
return action.data[0,0]
从这段代码中,我不断得到这样的错误:
TypeError: multinomial() missing 1 required positional arguments: "num_samples"
如果需要任何其他信息,将很快提供。
发布于 2018-12-06 04:34:20
基于documentation,您没有指定multinomial
函数的num_samples
来绘制多项式分布。
torch.multinomial(input,num_samples,replacement=False,out=None)返回张量,其中每一行都包含从张量input.
的相应行中的多项式概率分布中采样的num_samples索引
如下所示更改代码:
def select_action(self, state):
probs = F.softmax(self.model(Variable(state, volatile = True))*7)
action = probs.multinomial(1) # 1 is the number of samples to draw
return action.data[0,0]
https://stackoverflow.com/questions/53639839
复制相似问题