- 博客(2)
- 收藏
- 关注
原创 学习Faster R-CNN代码roi_pooling
转自:https://www.cnblogs.com/wind-chaser/p/11355002.html了解一下roi_pooling/src/roi_pooling.c中C代码:1.关于头文件TH/TH.h#include<TH/TH.h>包括了 pytorch C 代码数据结构和函数的声明,这是pytorch底层接口。2.roi_pooling_forward的参数1 int roi_pooling_forward(int pooled_height, int poo
2020-09-15 16:01:19 466
原创 关于pytorch中网络loss传播和参数更新的理解
转载自https://www.jb51.net/article/168009.htm1. loss计算和反向传播 1 2 3 4 5 6 7 8 import torch.nn as nn criterion = nn.MSELoss().cuda() output = model(input) loss = criterion(output, tar...
2020-09-03 17:22:11 3682
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人