flora

eval() self.eval()和self.train(False)等价 eval()在测试之前加，否则有输入数据即使不训练，它也会改变权值 pytorch会自己把BatchNormalization和DropOut固定住，不会取平均，而是用训练好的值。 train() model.train()让model变成训练模式 ...

首先我们来看一下nn和nn.functional有什么区别 nn.functional.xxx是函数接口，而nn.Xxx是nn.functional.xxx的类封装，并且nn.Xxx都继承于一个共同祖先nn.Module。这一点导致nn.Xxx除了具有nn.functional.xxx功能之外，内部附带了nn.Module相关的属性和方法，例如train(), eval(),load_state_...

转置 transpose和permute 对二维矩阵来说，转置就是把矩阵的行列互换 也就是原来是X[i][j]X[i][j]X[i][j]转置后变成X[j][i]X[j][i]X[j][i] torch.transpose transpose函数介绍 torch.transpose(input, dim0, dim1) → Tensor transpose就是把input 的第dim0维和di...

我们训练模型要进行批训练的时候，就涉及到每一批选取什么数据的问题，pytorch的DataLoader就帮我们包装了数据，它能帮我们有效迭代数据，这样就可以进行批训练。 pytorch 数据加载到模型的流程 pytorch 的数据加载到模型的操作顺序是这样的： ① 创建一个 Dataset 对象 ② 创建一个 DataLoader 对象 ③ 循环这个 DataLoader 对象，将img, lab...