摘要: title: libtorch教程(四) date: 2021 01 18 19:50:16 tags: libtorch 本章将详细介绍如何使用libtorch自带的数据加载模块,使用该模块是实现模型训练的重要条件。除非这个数据加载模块功能不够,不然继承libtorch的数据加载类还是很有必要的, 阅读全文
posted @ 2021-01-27 23:26 AllentBKY 阅读(2395) 评论(0) 推荐(1) 编辑
摘要: 基本模块搭建 模块化编程的思想非常重要,通过模块化编程可以大幅减少重复的敲代码过程,同时代码可读性也会增加。本章将讲述如何使用libtorch搭建一些MLP和CNN的基本模块。 MLP基本单元 首先是线性层的声明和定义,包括初始化和前向传播函数。代码如下: 在MLP的构造线性层模块类时,我们继承了t 阅读全文
posted @ 2021-01-27 23:23 AllentBKY 阅读(2753) 评论(0) 推荐(0) 编辑