- 博客(1)
- 收藏
- 关注
原创 PyTorch踩坑记录——torch.functional 与 torch.nn.functional的区别
问题描述:提示:刚入门深度学习,记录一些犯下的小错误:由于本周开始试图复现华为的CTR库以增加记忆,熟悉代码细节,没想到第一天看基础模块的时候就遇到了麻烦,在torch.utils类中,有如下获取损失函数的代码块:def get_loss_fn(loss): if isinstance(loss, str): if loss in ["bce", "binary_crossentropy", "binary_cross_entropy"]: loss =
2022-04-03 15:36:25 2477 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人