From 59b0533b06bc51ca2e865b273a735fc5b59c34ad Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 17 Jun 2020 18:45:24 -0600 Subject: [PATCH] Fix attimage step size --- codes/models/archs/SwitchedResidualGenerator_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SwitchedResidualGenerator_arch.py b/codes/models/archs/SwitchedResidualGenerator_arch.py index 4f55948d..76feccd2 100644 --- a/codes/models/archs/SwitchedResidualGenerator_arch.py +++ b/codes/models/archs/SwitchedResidualGenerator_arch.py @@ -129,7 +129,7 @@ class ConfigurableSwitchedResidualGenerator(nn.Module): if self.attentions: temp = max(1, int(self.init_temperature * (self.final_temperature_step - step) / self.final_temperature_step)) self.set_temperature(temp) - if step % 2 == 0: + if step % 250 == 0: [save_attention_to_image(experiments_path, self.attentions[i], self.transformation_counts[i], step, "a%i" % (i+1,), l_mult=float(self.transformation_counts[i]/4)) for i in range(len(self.switches))] def get_debug_values(self, step):