Pytorch
yyhhlancelot
心之所向,素履以往。
展开
-
Pytorch 关于nn.CrossEntropyLoss()与nn.BCEloss()以及nn.BCEWithLogitsLoss()的区别【坑】
最近用pytorch做实验,踩中一些坑,有小有大,这个问题花了我不少时间找到原因,姑且算个大坑。首先,这几个类分别对应的函数为:nn.CrossEntropyLoss() ——》nn.functional.cross_entropy()nn.BCEloss() ——》nn.functional.binary_cross_entropy()nn.BCEWithLogitsLoss() ——》...原创 2020-02-11 11:57:43 · 8957 阅读 · 5 评论 -
Pytorch中如何理解RNN LSTM GRU的input(重点理解seq_len / time_steps)
在建立时序模型时,若使用keras,我们在Input的时候就会在shape内设置好sequence_length(后面简称seq_len),接着便可以在自定义的data_generator内进行个性化的使用。这个值同时也就是time_steps,它代表了RNN内部的cell的数量,有点懵的朋友可以再去看看RNN的相关内容:个人总结:从RNN(内含BPTT以及梯度消失/爆炸)到 LSTM(内含GRU...原创 2020-01-15 16:17:03 · 8905 阅读 · 11 评论