Chainer和float64支持

时间:2019-06-17 07:56:37

标签: python chainer

我想知道如何使链接器支持float64精度计算。以下是我在链接器中编写的简单的自动编码器代码。最初我是用float32 dtype运行它的,但是现在我想测试一下float64的精度,我该怎么做?

class AutoEncoder(chainer.Chain):
    def __init__(self, num_input=1024, num_hidden=512, activate=F.leaky_relu, dropout_ratio=0.0):
        super(AutoEncoder, self).__init__(
            encoder=L.Linear(num_input, num_hidden),
            decoder=L.Linear(num_hidden, num_input))
        self.num_input = num_input
        self.num_hidden = num_hidden
        self.activate = activate
        self.dropout_ratio = dropout_ratio

    def __call__(self, x, hidden=False):
        h = F.dropout(self.activate(self.encoder(x)), ratio=self.dropout_ratio)
        if hidden:
            return h

        y = F.dropout((self.decoder(h)), ratio=self.dropout_ratio)
        return y

当我尝试使用float64数据运行它时,会引发dtype不匹配错误。

0 个答案:

没有答案