pytorch自编码网络AutoEncoder（pycharm动态图无法显示问题解决）

最新推荐文章于 2023-09-06 23:29:02 发布

小睿羊今天好好学习了吗

最新推荐文章于 2023-09-06 23:29:02 发布

阅读量378

点赞数

分类专栏： python pytorch 文章标签： pytorch 网络 pycharm

本文链接：https://blog.csdn.net/qq_41105449/article/details/117988793

版权

基于莫烦Python的课程，使用PyTorch 1.8.1实现自编码网络。在PyCharm中运行时遇到动态图无法显示的问题，原因是PyCharm的Python Scientific设置。通过取消勾选'Show plots in tool window'选项，可以解决动态图显示为乱码的问题。

摘要由CSDN通过智能技术生成

根据莫烦python课程在pytorch版本1.8.1上实现的。
莫烦python课程链接

import numpy as np
import torch
import torch.nn as nn
import torch.utils.data as Data
import torchvision
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 保证每次测试结果稳定
torch.manual_seed(0)

# 超参数
Epoch=6
batchsize=64
Lr=0.01
download_mnist=False
N_TEST_IMAGE=5

train_data=torchvision.datasets.MNIST(
    root='./mnist',
    train=True,
    transform=torchvision.transforms.ToTensor(),
    download=download_mnist,
)
train_loader=Data.DataLoader(dataset=train_data, batch_size=batchsize, shuffle=True)
class AutoEncoder(nn.Module):
    def __init__(self):
        super(AutoEncoder, self).__init__()
        self.encoder=nn.Sequential(
            nn.Linear(28*28, 128),
            nn.Tanh(),
            nn.Linear(128,64),
            nn.Tanh(),
            nn.Linear(64, 12),
            nn