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