05 2019 档案
摘要:numpy中有函数argsort来返回排序后的下标 pytorch中的torch.sort本身就能返回排序后的下标 Python中没有直接调用的接口,怎么办呢? 用enumerate再排序就可以了
阅读全文
摘要:Pytorch中的Tensor常用的类型转换函数(inplace操作): (1)数据类型转换 在Tensor后加 .long(), .int(), .float(), .double()等即可,也可以用.to()函数进行转换,所有的Tensor类型可参考https://pytorch.org/doc
阅读全文
摘要:model.train() :启用 BatchNormalization 和 Dropout model.eval() :不启用 BatchNormalization 和 Dropout 参考: https://pytorch.org/docs/stable/nn.html?highlight=mo
阅读全文