《使用 PyTorch 制作你的第一个生成对抗网络(GAN)》项目教程
2025-04-15 06:53:10作者:姚月梅Lane
1. 项目目录结构及介绍
本项目包含了使用PyTorch构建GAN(生成对抗网络)的详细教程,目录结构如下:
gan/
├── 00_pytorch_basics.ipynb
├── 01_pytorch_computation_graph.ipynb
├── 02_mnist_data.ipynb
├── 03_mnist_classifier.ipynb
├── 04_mnist_classifier_refinements.ipynb
├── 05_cuda_basics.ipynb
├── 06_gan_simple_pattern.ipynb
├── 07_gan_mnist_first_attempt.ipynb
├── 08_gan_mnist.ipynb
├── 09_gan_mnist_seed_experiments.ipynb
├── 10_celeba_download_make_hdf5.ipynb
├── 11_celeba_data.ipynb
├── 12_gan_celeba.ipynb
├── 13_cnn_mnist.ipynb
├── 14_gan_cnn_celeba.ipynb
├── 15_gan_cnn_celeba_refinements.ipynb
├── 16_cgan_mnist.ipynb
├── Appendix_B_generate.ipynb
├── Appendix_D_convergence.ipynb
├── LICENSE
├── README.md
文件说明:
00_pytorch_basics.ipynb:PyTorch基础教程。01_pytorch_computation_graph.ipynb:PyTorch计算图介绍。02_mnist_data.ipynb:MNIST数据集加载与预处理。03_mnist_classifier.ipynb:构建MNIST分类器。04_mnist_classifier_refinements.ipynb:MNIST分类器优化。05_cuda_basics.ipynb:CUDA基础使用。06_gan_simple_pattern.ipynb:简单的GAN模式示例。07_gan_mnist_first_attempt.ipynb:MNIST数据集上的GAN首次尝试。08_gan_mnist.ipynb:在MNIST数据集上实现GAN。09_gan_mnist_seed_experiments.ipynb:MNIST数据集GAN种子实验。10_celeba_download_make_hdf5.ipynb:CelebA数据集下载与转换为HDF5格式。11_celeba_data.ipynb:加载和预处理CelebA数据集。12_gan_celeba.ipynb:在CelebA数据集上实现GAN。13_cnn_mnist.ipynb:使用CNN构建MNIST分类器。14_gan_cnn_celeba.ipynb:使用CNN在CelebA数据集上实现GAN。15_gan_cnn_celeba_refinements.ipynb: CelebA数据集GAN优化。16_cgan_mnist.ipynb:条件GAN在MNIST数据集上的实现。Appendix_B_generate.ipynb:生成示例图像的附录。Appendix_D_convergence.ipynb:GAN收敛性研究附录。LICENSE:项目使用的GPL-2.0协议许可文件。README.md:项目说明文件。
2. 项目的启动文件介绍
项目的启动文件是00_pytorch_basics.ipynb,它提供了PyTorch的基础知识,是开始本项目前必须了解的内容。这个Jupyter笔记本文件将引导你了解PyTorch的基本概念,包括张量、自动微分等。
3. 项目的配置文件介绍
本项目没有特定的配置文件。所有的设置和参数都在各个Jupyter笔记本中通过代码直接配置。如果需要进行大规模训练或者想要调整模型参数,可以在相应的笔记本中修改代码中的参数设置。
在开始每个笔记本之前,请确保你已经安装了所需的库,并且设置了适当的环境变量。大多数配置都是通过代码注释或说明指出的,确保按照指示操作即可。
登录后查看全文
热门项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0423
源启盛夏_AtomGit暑期开发者成长计划「源启盛夏」暑期校园开发者成长计划旨在激活校园开源力量,通过积分激励、认证扶持、资源倾斜等形式,引导高校组织和开发者完成「入驻 — 建项目 — 做贡献 — 获认证 — 得资源」的完整闭环。无论你是想带领社团入驻平台的组织者,还是希望用代码贡献证明自己的开发者,都能在这里找到属于你的成长路径。Markdown00
jiuwenswarmJiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。Python0741
Hy3Hy3 是由腾讯混元团队研发的快慢思考融合的混合专家模型,总参数量 295B,激活参数 21B,MTP 层参数 3.8B。4 月底发布 Hy3 Preview 后,我们在 50 多个业务中获得了广泛的反馈,修复了各种体验问题,进一步提升了后训练的质量和规模。今天,我们发布 Hy3。它展现出显著强于同尺寸并比肩旗舰(参数规模往往是 Hy3 的 2~5 倍)开源模型的智能水平,显著提升了在各类产品和生产力任务中的实用价值。Python00
AscendNPU-IRAscendNPU-IR是基于MLIR(Multi-Level Intermediate Representation)构建的,面向昇腾亲和算子编译时使用的中间表示,提供昇腾完备表达能力,通过编译优化提升昇腾AI处理器计算效率,支持通过生态框架使能昇腾AI处理器与深度调优C++0298
PromptXPromptX · 领先的AI 智能体上下文平台 | PromptX · Leading AI Agent Context PlatformJavaScript05
项目优选
收起
暂无描述
Markdown
818
5.42 K
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
488
509
deepin linux kernel
C
32
16
Ascend Extension for PyTorch
Python
791
1.11 K
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
953
2.25 K
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
765
1.54 K
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.2 K
1.23 K
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
2.82 K
741
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
617
238
AscendNPU-IR是基于MLIR(Multi-Level Intermediate Representation)构建的,面向昇腾亲和算子编译时使用的中间表示,提供昇腾完备表达能力,通过编译优化提升昇腾AI处理器计算效率,支持通过生态框架使能昇腾AI处理器与深度调优
C++
415
298