- 博客(5)
- 收藏
- 关注
RSS订阅原创 for XX in dataloader和for XX in enumerate(dataloader)的区别
通过用MNIST数据集和CNN网络模型做实验得知： for i, inputs in train_loader:不加enumerate的话只能返回两个值，其中第一个值（这里是i）为输入的图片数据，第二个值为数据标签；for i, (inputs，labels) in enumerate（train_loader）: 加上enumerate，可以返回三个值，第一个值为序号，第二个值是输入数据，第三个值是数据标签。...
2022-02-25 09:14:01
6165
原创 pytorch 修改预训练model
class Net(nn.Module): def __init__(self , model): super(Net, self).__init__() #取掉model的后两层 self.resnet_layer = nn.Sequential(*list(model.children())[:-2]) self.transion_layer = nn.ConvTranspose2d(2048, 2048, kernel_siz..
2022-01-17 15:48:34
336
原创 No module named ‘yaml‘
在使用如下命令时python3 teacher_training.py遇到了Traceback (most recent call last): File "teacher_training.py", line 9, in <module> from AnomalyNet import AnomalyNet File "/home/lhy/liuwork/src/AnomalyNet.py", line 2, in <module> import
2022-01-17 15:27:13
5449
原创 图像数组索引
1.三维数组的索引参考：( NumPy 三维数组以及高维数组切片和索引_king52113141314的博客-优快云博客_numpy三维数组索引https://blog.youkuaiyun.com/king52113141314/article/details/108405364?spm=1001.2014.3001.5506三维数组的索引中，第一个维度是矩阵个数，第二个维度是行，第三个维度是列。关于三维数组的[::-1]的倒序操作：以下为例：可以看到，将[::-1]放在第一维...
2021-12-22 19:03:52
1793
原创 IndexError: index 16 is out of bounds for dimension 0 with size 16
通过打印labels和predicted的数据发现，是因为batchsize分批不平均，导致predicted最后有一组只有16个数据。而range（）超过了16，所以报错。解决办法：1.改变batchsize，使数据集分组刚刚好 2.填充最后一组不足的数据...
2021-10-20 16:55:14
7595
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人