- 博客(7)
- 收藏
- 关注
原创 刷力扣记录-二叉搜索树
还在[代码随想录]()的二叉搜索树的部分,有段日子没有刷了,发现不刷也没啥干的,无聊真是可怕,还是开始起来。这次的话题是[501. 二叉搜索树中的众数](),主要的思想就是怎么得到一个众数。
2023-05-26 18:09:12
87
1
原创 在Anaconda虚拟环境中安装cuda和cudnn
这个方法可以避免安装多个cuda的切换问题,直接在每个虚拟环境中安装自己需要的cuda版本,相比较tensorflow官方推荐的docker的安装方法,可以达到相同的效果,但是实践起来更简单,总之推荐大家以后在虚拟环境下安装所需要的cuda和cudnn。
2023-05-26 13:59:30
3069
1
原创 我对pytorch 中Dataset的理解
一两个类 datasets 和 DataLoader函数的完整表达是 torch.utils.data.Dataset和torch.utils.data.DataLoader。如果纯粹加载tensor或者image数据可以直接使用TensorDataset和ImageFolder进行。当使用自己的数据集,避免不了对dataset类进行改写来使得dataloader进行数据的读取,该类是建立索引到样本的一个映射。主要有3个方法,类的初始化方法__init__(),类内容构建方法__getitem__(),
2022-04-02 23:11:29
1166
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人