Pytorch自定义参数层
摘要：
注意，一般官方接口都带有可导功能，如果你实现的层不具有可导功能，就需要自己实现梯度的反向传递。 阅读全文
posted @ 2019-07-18 16:02 那抹阳光1994 阅读(3076) 评论(0) 推荐(0) 编辑
pytorch BCELoss和BCEWithLogitsLoss
摘要：
BCELoss CLASS torch.nn.BCELoss(weight=None, size_average=None, reduce=None, reduction='mean') 创建一个标准来度量目标和输出之间的二进制交叉熵。 unreduced (i.e. with reduction 阅读全文
posted @ 2019-07-18 15:54 那抹阳光1994 阅读(14767) 评论(0) 推荐(0) 编辑
