Bonky
Neither beliver nor reject anything, because any other person has rejected of believed it. Heaven has given you a mind for judging truth and error, Use it.
By Thomas Jefferson

训练神经网络的一些技巧(1)

view函数

主要用法如下,参数中的-1代表补全维度

随机选择 batch

可以利用random.choice(a) 函数,从a 中随机选取一个。下面可以作为一个模板,当你需要选取一个 batchsize 的训练数据时

让一个 batch 所有的向量补全到相同长度

使用以下代码,输入一个 batch

具体zip_longest 使用方法如下

保存模型的状态

pytorch 提供了方法 torch.save()torch.load() 供我们保存和加载模型

其中obj 可以是任何东西,比如词典,类,Tensor,下面是一个实际例子,可以作为模板写入代码:

logging 记录训练过程

在以前如果不用 loggingbug 我都是用的 debug,但是训练过程数据量大,然后变量维度也大,一步一步调谁顶得住啊,所以可以用 logging 记录训练的过程。

首先先引入包,设置配置

对于你要记录的信息,根据下面的例子来改动:

Share

You may also like...

发表评论

电子邮件地址不会被公开。 必填项已用*标注