《使用 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 StartedRust0117- DDeepSeek-V4-ProDeepSeek-V4-Pro(总参数 1.6 万亿,激活 49B)面向复杂推理和高级编程任务,在代码竞赛、数学推理、Agent 工作流等场景表现优异,性能接近国际前沿闭源模型。Python00
MiMo-V2.5-ProMiMo-V2.5-Pro作为旗舰模型,擅⻓处理复杂Agent任务,单次任务可完成近千次⼯具调⽤与⼗余轮上 下⽂压缩。Python00
GLM-5.1GLM-5.1是智谱迄今最智能的旗舰模型,也是目前全球最强的开源模型。GLM-5.1大大提高了代码能力,在完成长程任务方面提升尤为显著。和此前分钟级交互的模型不同,它能够在一次任务中独立、持续工作超过8小时,期间自主规划、执行、自我进化,最终交付完整的工程级成果。Jinja00
SenseNova-U1-8B-MoT-SFTenseNova U1 是一系列全新的原生多模态模型,它在单一架构内实现了多模态理解、推理与生成的统一。 这标志着多模态AI领域的根本性范式转变:从模态集成迈向真正的模态统一。SenseNova U1模型不再依赖适配器进行模态间转换,而是以原生方式在语言和视觉之间进行思考与行动。Python00
MiniMax-M2.7MiniMax-M2.7 是我们首个深度参与自身进化过程的模型。M2.7 具备构建复杂智能体应用框架的能力,能够借助智能体团队、复杂技能以及动态工具搜索,完成高度精细的生产力任务。Python00
热门内容推荐
最新内容推荐
项目优选
收起
暂无描述
Dockerfile
718
4.58 K
Ascend Extension for PyTorch
Python
583
719
deepin linux kernel
C
28
16
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
965
960
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
419
364
Claude 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 Started
Rust
750
117
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
1.63 K
956
昇腾LLM分布式训练框架
Python
154
180
Oohos_react_native
React Native鸿蒙化仓库
C++
342
390
暂无简介
Dart
957
238