(1)进行了函数内容缩进(2)加上了alpha前面忘加的self.(3)loss的求和方式增加了一种batchsize方式 #21
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
您的代码真的非常好,简洁易懂,教程详细,非常感谢您的无私分享,但是我运行时注意到代码尚且存在两点小问题【(1)与(2)】所以特此发起了Pull 请求
(1)def forward(self, preds, labels):函数中的内容您忘了缩进了
(2)第45行: alpha = self.alpha.to(preds.device)忘记在alpha前面加上self了,使用cuda时会报错--alpha和label不在一个设备上。
(3)原先的loss返回方式只有sum和mean两种,我新增加了一种batch_size方式。