摘要:
1. torch.no_grad (1) 用法 with torch.no_grad(): 具体操作 (2) 说明 上例的“具体操作”中均不更新梯度,这样可以节约计算时间和内存。一般用于验证或者测试阶段。 2. param.requires_grad (1)... 阅读全文
摘要:
下例使用torchvision库提取了resnet最后一层的卷积特征;resnet各block的卷积特性,以及金字塔特性。 具体取哪一层特征视使用场景而定,resnet各block的输出包含更丰富的特征;从resnet最后一层提取的特征更为抽象;fpn每层通道... 阅读全文