From f41b09ef7fa6fd5626531275f2770db0ec410cad Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sun, 8 Aug 2021 17:28:57 +0800 Subject: [PATCH] dbg --- ppcls/arch/backbone/base/theseus_layer.py | 2 ++ ppcls/arch/backbone/legendary_models/vgg.py | 1 - ppcls/engine/trainer.py | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/base/theseus_layer.py b/ppcls/arch/backbone/base/theseus_layer.py index 41599cd9..04871e48 100644 --- a/ppcls/arch/backbone/base/theseus_layer.py +++ b/ppcls/arch/backbone/base/theseus_layer.py @@ -109,6 +109,8 @@ class WrapLayer(TheseusLayer): for return_pattern in return_patterns: if re.match(return_pattern, layer_name): self.sub_layer._sub_layers[layer_i].res_dict = self.res_dict + self._sub_layers[layer_i].register_forward_post_hook( + self._sub_layers[layer_i]._save_sub_res_hook) if isinstance(self.sub_layer._sub_layers[layer_i], TheseusLayer): self.sub_layer._sub_layers[layer_i].update_res(return_patterns) diff --git a/ppcls/arch/backbone/legendary_models/vgg.py b/ppcls/arch/backbone/legendary_models/vgg.py index fbfdaca0..c8a16932 100644 --- a/ppcls/arch/backbone/legendary_models/vgg.py +++ b/ppcls/arch/backbone/legendary_models/vgg.py @@ -137,7 +137,6 @@ class VGGNet(TheseusLayer): self.fc1 = Linear(7 * 7 * 512, 4096) self.fc2 = Linear(4096, 4096) self.fc3 = Linear(4096, class_num) - self.update_res(return_patterns) def forward(self, inputs, res_dict=None): x = self.conv_block_1(inputs) diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py index eb588ed0..0fcb9c32 100644 --- a/ppcls/engine/trainer.py +++ b/ppcls/engine/trainer.py @@ -77,6 +77,7 @@ class Trainer(object): self.model = build_model(self.config["Arch"]) if "return_patterns" in self.config["Arch"] and isinstance(self.model, TheseusLayer): + self.model.update_res(self.config["Arch"]["return_patterns"]) self.return_inter = True else: self.return_inter = False -- GitLab