05 2019 档案

摘要:numpy中有函数argsort来返回排序后的下标 pytorch中的torch.sort本身就能返回排序后的下标 Python中没有直接调用的接口,怎么办呢? 用enumerate再排序就可以了 阅读全文
posted @ 2019-05-30 19:32 sbj123456789 阅读(9394) 评论(0) 推荐(1) 编辑
摘要:Pytorch中的Tensor常用的类型转换函数(inplace操作): (1)数据类型转换 在Tensor后加 .long(), .int(), .float(), .double()等即可,也可以用.to()函数进行转换,所有的Tensor类型可参考https://pytorch.org/doc 阅读全文
posted @ 2019-05-09 16:01 sbj123456789 阅读(81120) 评论(0) 推荐(1) 编辑
摘要:model.train() :启用 BatchNormalization 和 Dropout model.eval() :不启用 BatchNormalization 和 Dropout 参考: https://pytorch.org/docs/stable/nn.html?highlight=mo 阅读全文
posted @ 2019-05-07 15:23 sbj123456789 阅读(37677) 评论(0) 推荐(2) 编辑

点击右上角即可分享
微信分享提示