[トピック内 4 記事 (1 - 4 表示)] << 0 >>
返信でテキストボックスみたら、そう書いてありましたね。 ■No93299 (くまふぁん さん) に返信 > class AAA: > def __init__(self, model, feature_layer): > > def SFG(module, in_grad, out_grad): > self.feature_grad = out_grad[0] > self.hooks.append(self.feature_layer.register_backward_hook(SFG)) このため、SFG は直下の self.feature_grad = out_grad[0] だけの処理を持つ、ローカル関数です。
管理者用
- Child Tree -