- 博客(5)
- 收藏
- 关注
RSS订阅原创 [Pytorch入门] 土堆课程笔记
创建dataset数据集，分为训练集和测试集用dataloader进行数据加载用len（）查看数据集长度，比如cifar10 训练集和测试集各有多少张图片自己写的模型model model文件和train文件要在同一个目录下，from model import *写损失函数 loss选择合适的优化器设置网络训练参数调用haha.train()让网络进入训练状态（传数据，过模型得到output，loss，梯度置零，反向传播，打印~）
2023-11-24 17:26:05
860
1
原创 Transformer学习笔记
从很多方面介绍了Transformer的架构，也写了很多基础模块：Encoder层、Decoder层、Embedding层+位置编码、残差结构等等，还是比较详细的，适合新手哈哈哈
2023-05-12 22:31:40
204
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人