- 博客(5)
- 收藏
- 关注
原创 Pytorch中自定义数据集
在Pytorch中,自定义数据集一般通过继承 torch.utils.data.Dataset 类来实现,一般需要定义三个magic methods ::初始化新创建的对象。:返回容器类型的长度。:获取序列的元素。lmdb_path,return envelse:connect_db函数:数据集存储形式为LMDB,如果save_to_self为False,那数据不保存在类实例中,如果save_to_self为True,那么数据保存在实例中。
2024-10-24 19:20:50 340
原创 python中从kwargs中提取参数
方法1:使用dich.pop()方法,可以提取并移除字典中的键值对。方法二:直接访问,但不会移除键值对。
2024-10-23 20:54:23 70 1
原创 关于transformers中BartTokenizer模块使用
BartTokenizer模块是transformers中用于处理BART模型的分词器,在自然语言处理中起到了很重要的作用,其任务是将输入的文本进行分词,分解为模型可以理解的token,并且可以将token转换为对应的ID,以便进行模型输入,及编码和解码。本次测试采用的预训练的模型为AI4S Cup 基于红外光谱的分子识别模型,分解SMILE表达。: 注意力掩码,指示哪些 token 是填充的。: 编码后的输入 ID,通常是一个二维数组。进行分词、编码、解码测试。
2024-10-23 20:33:14 115
原创 STM32+Keilv6+cubemx进行FPU运算
选择DSP Library 下载DSP Library,此步非常重要,若没有此步,cubemx不会生成arm_cortexM4lf_math.lib。点击ok,选择如图所示的位置,勾选DSP Library,此时就可以生成代码了。选择show or hide the filter panel。单片机型号:stm32f407zgt6。选择select components。打开魔术棒,添加以下定义。2.keilv6配置。
2024-03-16 02:42:33 558 2
原创 利用Anaconda+Pycharm创建Pytorch环境
这里是因为之前创建pytorch环境遇到了非常多的问题,这次重新创建一个pytorch的环境,用于以后如果需要重新安装,则根据这篇博客查考,起到一个参考博客的作用。
2024-01-26 16:18:08 492 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人