no need to store prototype local model trace

pyro-ppl · Jun 29, 2023 · 063b604 · 063b604
1 parent cbc7b3b
commit 063b604
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/numpyro/infer/autoguide.py b/numpyro/infer/autoguide.py
@@ -1259,10 +1259,6 @@ def _setup_prototype(self, *args, **kwargs):
                 self.local_guide(*local_args, **local_kwargs)
             self.prototype_local_guide_trace = tr
 
-        with handlers.block(), handlers.trace() as tr, handlers.seed(rng_seed=0):
-            self.local_model(*local_args, **local_kwargs)
-        self.prototype_local_model_trace = tr
-
     def __call__(self, *args, **kwargs):
         if self.prototype_trace is None:
             # run model to inspect the model structure