- 博客(3)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 安装albumentations-imgaug-pycocotools-conda install -c akode gym
conda install -c simpleitk simpleitk 安装skimage conda install scikit-image 安装albumentations conda install albumentations -c albumentations
2020-01-20 11:09:06
3034
原创 pytorch load函数中cpu与gpu转化
cpu -> cpu或者gpu -> gpu: checkpoint = torch.load(‘modelparameters.pth’) model.load_state_dict(checkpoint) 2. cpu -> gpu 1 torch.load(‘modelparameters.pth’, map_location=lambda storage, loc: ...
2020-01-19 17:17:06
515
原创 torch.nn.functional.normalize
torch normalize torch.nn.functional.normalize 范数 0范数，1范数，2范数 见范数blog： https://blog.csdn.net/cursethegod/article/details/80372718 见示例： input_ = torch.randn((3, 4)) a = torch.nn.Softmax()(input_) b = t...
2020-01-02 14:40:25
507
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人