pytorch IntermediateLayerGetter

最新推荐文章于 2023-11-26 16:05:15 发布

AI算法网奇

最新推荐文章于 2023-11-26 16:05:15 发布

阅读量6.8k

点赞数 1

分类专栏： python宝典 pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/102475974

版权

python宝典同时被 2 个专栏收录

1012 篇文章 368 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

pytorch知识宝典

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

IntermediateLayerGetter

torch是能看源码的，可以直接找。

官方源码：

from collections import OrderedDict

import torch
from torch import nn


class IntermediateLayerGetter(nn.ModuleDict):
    """
    Module wrapper that returns intermediate layers from a model

    It has a strong assumption that the modules have been registered
    into the model in the same order as they are used.
    This means that one should **not** reuse the same nn.Module
    twice in the forward if you want this to work.

    Additionally, it is only able to query submodules that are directly
    assigned to the model. So if `model` is passed, `model.feature1` can
    be returned, but not `model.feature1.layer2`.

了解本专栏