首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >为什么在使用Deep Q learning时会出现此错误?

为什么在使用Deep Q learning时会出现此错误?
EN

Stack Overflow用户
提问于 2018-12-06 03:57:12
回答 1查看 1.8K关注 0票数 4

我一直在Windows10机器上使用Deep Q Learning。我有0.4.1版本的pytorch和NVIDA显卡。

代码语言:javascript
复制
def select_action(self, state):
    probs = F.softmax(self.model(Variable(state, volatile = True))*7)
    action = probs.multinomial()
    return action.data[0,0]

从这段代码中,我不断得到这样的错误:

代码语言:javascript
复制
TypeError: multinomial() missing 1 required positional arguments: "num_samples"

如果需要任何其他信息,将很快提供。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-12-06 04:34:20

基于documentation,您没有指定multinomial函数的num_samples来绘制多项式分布。

torch.multinomial(input,num_samples,replacement=False,out=None)返回张量,其中每一行都包含从张量input.

的相应行中的多项式概率分布中采样的num_samples索引

如下所示更改代码:

代码语言:javascript
复制
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]
票数 4
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/53639839

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档