- 博客(2)
- 收藏
- 关注
RSS订阅原创 Pytorch自定义损失函数losses.py
PyTorch已经有很多标准的损失函数，有时也需要创建自己的损失函数。为此，需要创建一个单独的文件 losses.py ，然后扩展 nn.Module类创建自定义损失函数：class CustomLoss(torch.nn.Module): def __init__(self): super(CustomLoss,self).__init__() ...
2019-05-31 15:20:45
5128
原创 Pytorch使用技巧
import 包模块import argparseimport osimport torchfrom torch import nn, optimfrom torch.nn import functional as Ffrom torch.utils.data import DataLoaderfrom torchvision import datasets, transform...
2019-05-31 13:29:05
522
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人