コード例 #1
0
ファイル: conv_filter_pruning.py プロジェクト: unixnme/webdnn
    def __init__(self, base: Convolution2D, limit: int, removed: int = 0):
        if base.has_attribute(ConvFilterPruned):
            raise ValueError(
                f"\'ConvFilterPruned\' attribute has been already registered to {base}."
            )

        super(ConvFilterPruned, self).__init__(base)
        self.limit = limit  # type: int
        self.removed = removed  # type: int
コード例 #2
0
ファイル: conv_filter_pruning.py プロジェクト: unixnme/webdnn
 def get(base: Convolution2D):
     if base.has_attribute(ConvFilterPruned):
         return base.get_attribute(ConvFilterPruned)[0]
コード例 #3
0
ファイル: conv_filter_pruning.py プロジェクト: unixnme/webdnn
 def has(base: Convolution2D):
     return base.has_attribute(ConvFilterPruned)