美文网首页pytorch
李沐 d2l在pycharm上无法显示动图

李沐 d2l在pycharm上无法显示动图

作者: 午字横 | 来源:发表于2022-11-15 09:48 被阅读0次

    解决代码的办法:
    在d2l.torch中的Class Animator类中的add函数的倒数第二行加上一句
    plt.draw();plt.pause(0.001)。
    jupyter是每次会自动调用图像显示函数的,所以源码里面没有但是在jupyter中可以显示。

     def add(self, x, y):
            # Add multiple data points into the figure
            if not hasattr(y, "__len__"):
                y = [y]
            n = len(y)
            if not hasattr(x, "__len__"):
                x = [x] * n
            if not self.X:
                self.X = [[] for _ in range(n)]
            if not self.Y:
                self.Y = [[] for _ in range(n)]
            for i, (a, b) in enumerate(zip(x, y)):
                if a is not None and b is not None:
                    self.X[i].append(a)
                    self.Y[i].append(b)
            self.axes[0].cla()
            for x, y, fmt in zip(self.X, self.Y, self.fmts):
                self.axes[0].plot(x, y, fmt)
            self.config_axes()
            display.display(self.fig)
            plt.draw();plt.pause(0.001)
            display.clear_output(wait=True)
    
    image.png

    相关文章

      网友评论

        本文标题:李沐 d2l在pycharm上无法显示动图

        本文链接:https://www.haomeiwen.com/subject/samsxdtx.html